Spaces:

thejagstudio
/

NanoDiffusion

Sleeping

thejagstudio commited on 17 days ago

Commit

76cfc68

1 Parent(s): c27ef68

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -302,7 +302,7 @@ v1_model = V1_GPT(v1_config)
 try:
     v1_model.load_state_dict(
         torch.hub.load_state_dict_from_url(
-            'https://huggingface.co/spaces/thejagstudio/diffusion-gpt/resolve/main/final_model.pth?download=true',
             map_location=device
         )
     )

 try:
     v1_model.load_state_dict(
         torch.hub.load_state_dict_from_url(
+            'https://huggingface.co/spaces/thejagstudio/NanoDiffusion/resolve/main/final_model.pth?download=true',
             map_location=device
         )
     )