Spaces:

TharunSiva
/

Mini-GPT

Runtime error

TharunSiva commited on Oct 24, 2023

Commit

e20d7c3

1 Parent(s): 7d906de

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from model import *
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = GPTLanguageModel().to(device)
-model.load_state_dict(torch.load("mini-gpt.pth"))
 answer = decode(model.generate(context, max_new_tokens=1000)[0].tolist())

 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = GPTLanguageModel().to(device)
+model.load_state_dict(torch.load("mini-gpt.pth",map_location=DEVICE), strict=False)
 answer = decode(model.generate(context, max_new_tokens=1000)[0].tolist())