Spaces:

yyfz233
/

Pi3

Running on Zero

yyfz233 commited on Jul 16

Commit

1a5d31c

1 Parent(s): 853528a

Fix bug for checkpoint loadding

Files changed (1) hide show

app.py CHANGED Viewed

@@ -563,9 +563,9 @@ if __name__ == '__main__':
     print("Initializing and loading Pi3 model...")
     model = Pi3()
-    # _URL = "https://huggingface.co/yyfz233/Pi3/resolve/main/model.safetensors"
-    # model.load_state_dict(torch.hub.load_state_dict_from_url(_URL))
-    model.load_state_dict(torch.load('ckpts/pi3.pt', weights_only=False, map_location=device))
     model.eval()
     model = model.to(device)

     print("Initializing and loading Pi3 model...")
     model = Pi3()
+    _URL = "https://huggingface.co/yyfz233/Pi3/resolve/main/model.safetensors"
+    model.load_state_dict(torch.hub.load_state_dict_from_url(_URL))
+    # model.load_state_dict(torch.load('ckpts/pi3.pt', weights_only=False, map_location=device))
     model.eval()
     model = model.to(device)