Update dalle/utils/utils.py

dalle/utils/utils.py

@@ -55,16 +55,16 @@ def download(url: str, root: str) -> str:
     #if os.path.isfile(download_target) and (os.path.exists(result_path) and not os.path.isfile(result_path)):
         #return result_path
 
-    # with urllib.request.urlopen(url) as source, open(download_target, 'wb') as output:
-    #     with tqdm(total=int(source.info().get('Content-Length')), ncols=80, unit='iB', unit_scale=True,
-    #               unit_divisor=1024) as loop:
-    #         while True:
-    #             buffer = source.read(8192)
-    #             if not buffer:
-    #                 break
-
-    #             output.write(buffer)
-    #             loop.update(len(buffer))
+    with urllib.request.urlopen(url) as source, open(download_target, 'wb') as output:
+        with tqdm(total=int(source.info().get('Content-Length')), ncols=80, unit='iB', unit_scale=True,
+                  unit_divisor=1024) as loop:
+            while True:
+                buffer = source.read(8192)
+                if not buffer:
+                    break
+
+                output.write(buffer)
+                loop.update(len(buffer))
 
     # if hashlib.md5(open(download_target, 'rb').read()).hexdigest() != expected_md5:
         # raise RuntimeError(f'Model has been downloaded but the md5 checksum does not not match')