Spaces:

Skywork
/

UniPic

Runtime error

yichenchenchen commited on Jul 30, 2025

Commit

14ebc1a

verified ·

1 Parent(s): aa0a0cc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 #model_path = "Skywork/Skywork-UniPic-1.5B"
 from huggingface_hub import snapshot_download
+model_dir = snapshot_download(repo_id="Skywork/Skywork-UniPic-1.5B")
+model_path = os.path.join(model_dir,"pytorch_model.bin")
 ckpt_name = "UniPic"
 inferencer = Inferencer(