Spaces:

scy639
/

UniBioTransfer

Running on Zero

scy639 commited on Apr 23

Commit

da5f595

verified ·

1 Parent(s): 4f7f7e1

Update hf_model.py

Files changed (1) hide show

hf_model.py CHANGED Viewed

@@ -150,7 +150,7 @@ class UniBioTransferModel(LatentDiffusion, PyTorchModelHubMixin):
             _download_file(repo_id, "checkpoints/pretrained.json", cache_dir)
         if download_sd14 and not sd14_path.exists():
-            _download_file(SD14_REPO, SD14_FILENAME, cache_dir)
         if download_deps:
             if not arcface_path.exists():

             _download_file(repo_id, "checkpoints/pretrained.json", cache_dir)
         if download_sd14 and not sd14_path.exists():
+            _download_file(SD14_REPO, SD14_FILENAME, cache_dir/"checkpoints")
         if download_deps:
             if not arcface_path.exists():