Spaces:

Ruurd
/

tini

Running on Zero

Ruurd commited on 25 days ago

Commit

0ffa4b5

verified ·

1 Parent(s): 939752c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,8 @@ def load_model():
     ckpt_path = hf_hub_download(
         repo_id="ruurd/tini_model",
         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_model",
         filename="diffusion-model.pth",
+        token=os.getenv("HF_TOKEN"),
+        revision="8bb2d44"
     )
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")