Spaces:

JackAILab
/

ConsistentID

Running on Zero

JackAILab commited on May 8, 2024

Commit

31287b1

verified ·

1 Parent(s): a6ea34d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ print(zero.device) # <-- 'cpu' 🤔
 script_directory = os.path.dirname(os.path.realpath(__file__))
 # download ConsistentID checkpoint to cache
-base_model_path = hf_hub_download(repo_id="SG161222/Realistic_Vision_V6.0_B1_noVAE")
 consistentID_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="ConsistentID-v1.bin", repo_type="model")
 ### Load base model

 script_directory = os.path.dirname(os.path.realpath(__file__))
 # download ConsistentID checkpoint to cache
+base_model_path = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 consistentID_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="ConsistentID-v1.bin", repo_type="model")
 ### Load base model