Changed to 3B
Browse files
app.py
CHANGED
|
@@ -184,7 +184,7 @@ if is_running_on_spaces():
|
|
| 184 |
# Load from Hugging Face Hub
|
| 185 |
ckpt_path = hf_hub_download(
|
| 186 |
repo_id="ruurd/tini_model",
|
| 187 |
-
filename="diffusion-model-
|
| 188 |
token=os.getenv("HF_TOKEN")
|
| 189 |
)
|
| 190 |
else:
|
|
|
|
| 184 |
# Load from Hugging Face Hub
|
| 185 |
ckpt_path = hf_hub_download(
|
| 186 |
repo_id="ruurd/tini_model",
|
| 187 |
+
filename="diffusion-model-3B.pth",
|
| 188 |
token=os.getenv("HF_TOKEN")
|
| 189 |
)
|
| 190 |
else:
|