Spaces:
Running
on
Zero
Running
on
Zero
Fix model repo name
Browse files
app.py
CHANGED
@@ -36,10 +36,10 @@ assistant_marker_ids = tokenizer.encode("Assistant:", add_special_tokens=False)
|
|
36 |
|
37 |
def load_model():
|
38 |
ckpt_path = hf_hub_download(
|
39 |
-
repo_id="ruurd/
|
40 |
filename="diffusion-model.pth",
|
41 |
token=os.getenv("HF_TOKEN"),
|
42 |
-
# revision="
|
43 |
)
|
44 |
|
45 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
36 |
|
37 |
def load_model():
|
38 |
ckpt_path = hf_hub_download(
|
39 |
+
repo_id="ruurd/tini_model",
|
40 |
filename="diffusion-model.pth",
|
41 |
token=os.getenv("HF_TOKEN"),
|
42 |
+
# revision="xxx",
|
43 |
)
|
44 |
|
45 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|