Hugging Face's logo

Spaces:
to0ony
/
final-thesis-plotgen-app
Sleeping

App Files Community
to0ony commited on
Commit
b4fee78
·
1 Parent(s): 08585a9

fixed torch loading

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def load_model():
29
  gcfg.n_embd = int(cfg["n_embd"])
30
 
31
  model = GPT(gcfg)
32
- sd = torch.load(mdl_path, map_location="cpu")
33
  model.load_state_dict(sd, strict=True)
34
  model.to(DEVICE)
35
  model.eval()
 
29
  gcfg.n_embd = int(cfg["n_embd"])
30
 
31
  model = GPT(gcfg)
32
+ sd = torch.load(mdl_path, map_location="cpu", weights_only=False)
33
  model.load_state_dict(sd, strict=True)
34
  model.to(DEVICE)
35
  model.eval()