Spaces:

Ruurd
/

lad

Running on Zero

Ruurd commited on Jun 5

Commit

7ab60eb

verified ·

1 Parent(s): 86c363a

Changed to 3B

Files changed (1) hide show

app.py CHANGED Viewed

@@ -184,7 +184,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: