Spaces:

Ruurd
/

tini

Sleeping

Ruurd commited on Apr 15

Commit

7287d81

verified ·

1 Parent(s): a73a13e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -43,7 +43,8 @@ def load_model():
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_bi",
         filename="diffusion-model.pth",
-        token=os.getenv("HF_TOKEN")
     )
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_bi",
         filename="diffusion-model.pth",
+        token=os.getenv("HF_TOKEN"),
+        revision="5a22a8b6168466dbbf704efd00d8cbf2eee51426",
     )
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")