Spaces:

to0ony
/

final-thesis-plotgen-app

Runtime error

to0ony commited on Aug 30, 2025

Commit

520c6fd

1 Parent(s): b4fee78

fixed torch loading

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ def load_model():
     model = GPT(gcfg)
     sd = torch.load(mdl_path, map_location="cpu", weights_only=False)
-    model.load_state_dict(sd, strict=True)
     model.to(DEVICE)
     model.eval()

     model = GPT(gcfg)
     sd = torch.load(mdl_path, map_location="cpu", weights_only=False)
+    model.load_state_dict(sd["model_state_dict"], strict=True)
     model.to(DEVICE)
     model.eval()