Spaces:
Sleeping
Sleeping
Commit
·
59bc37f
1
Parent(s):
b97e59a
Minor chnage
Browse files
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))
|
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")
|