Spaces:

AkashDataScience
/

GPT-2

Sleeping

AkashDataScience commited on Jul 3, 2024

Commit

467a29f

1 Parent(s): 78f0cbc

Changed model name

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
     device = "mps"
 model = GPT(GPTConfig())
-model.load_state_dict(torch.load("nanogpt.pth", map_location=torch.device(device)), strict=False)
 model.to(device)

     device = "mps"
 model = GPT(GPTConfig())
+model.load_state_dict(torch.load("gpt2.pt", map_location=torch.device(device)), strict=False)
 model.to(device)