Spaces:

Ruurd
/

tini

Running on Zero

Ruurd commited on 8 days ago

Commit

cea9bb0

verified ·

1 Parent(s): 31e34c4

Change to 3B

Files changed (1) hide show

app.py CHANGED Viewed

@@ -183,7 +183,7 @@ if is_running_on_spaces():
     # Load from Hugging Face Hub
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_model",
-        filename="diffusion-model-8B.pth",
         token=os.getenv("HF_TOKEN")
     )
 else:

     # Load from Hugging Face Hub
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_model",
+        filename="diffusion-model-3B.pth",
         token=os.getenv("HF_TOKEN")
     )
 else: