Spaces:
Sleeping
Sleeping
Commit Β·
76cfc68
1
Parent(s): c27ef68
Update app.py
Browse files
app.py
CHANGED
|
@@ -302,7 +302,7 @@ v1_model = V1_GPT(v1_config)
|
|
| 302 |
try:
|
| 303 |
v1_model.load_state_dict(
|
| 304 |
torch.hub.load_state_dict_from_url(
|
| 305 |
-
'https://huggingface.co/spaces/thejagstudio/
|
| 306 |
map_location=device
|
| 307 |
)
|
| 308 |
)
|
|
|
|
| 302 |
try:
|
| 303 |
v1_model.load_state_dict(
|
| 304 |
torch.hub.load_state_dict_from_url(
|
| 305 |
+
'https://huggingface.co/spaces/thejagstudio/NanoDiffusion/resolve/main/final_model.pth?download=true',
|
| 306 |
map_location=device
|
| 307 |
)
|
| 308 |
)
|