AkashDataScience commited on
Commit
59bc37f
·
1 Parent(s): b97e59a

Minor chnage

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
12
  device = "mps"
13
 
14
  model = GPT(GPTConfig())
15
- model.load_state_dict(torch.load("gpt2.pt", map_location=torch.device(device)), strict=False)
16
 
17
  model.to(device)
18
  model = torch.compile(model, fullgraph=True, backend="cudagraphs")
 
12
  device = "mps"
13
 
14
  model = GPT(GPTConfig())
15
+ model.load_state_dict(torch.load("gpt2.pt", map_location=torch.device(device)))
16
 
17
  model.to(device)
18
  model = torch.compile(model, fullgraph=True, backend="cudagraphs")