UniPic

Running on Zero

yichenchenchen commited on 21 days ago

Commit

38061b4

verified ·

1 Parent(s): 587c0d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ from accelerate.utils import set_seed
 #model_path = "Skywork/Skywork-UniPic-1.5B"
 from huggingface_hub import hf_hub_download
-model_path = hf_hub_download(repo_id="Skywork/Skywork-UniPic-1.5B", filename="pytorch_model.bin")
 ckpt_name = "UniPic"
 inferencer = Inferencer(

 #model_path = "Skywork/Skywork-UniPic-1.5B"
 from huggingface_hub import hf_hub_download
+model_path = hf_hub_download(repo_id="Skywork/Skywork-UniPic-1.5B")
 ckpt_name = "UniPic"
 inferencer = Inferencer(