Spaces:

nickgardner
/

chatbot-demo

Sleeping

nickgardner commited on Apr 22, 2023

Commit

952f87e

•

1 Parent(s): 8d7ed76

see if model weights need torch load

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,8 +29,8 @@ N = 6
 src_vocab = len(vocab)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = Transformer(len(vocab), len(vocab), d_model, N, heads).to(device)
-model.load_state_dict(hf_hub_download(repo_id="nickgardner/chatbot",
-                                      filename="alpaca_train_380_epoch.pt"))
 model.eval()
 def respond(custom_string):

 src_vocab = len(vocab)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = Transformer(len(vocab), len(vocab), d_model, N, heads).to(device)
+model.load_state_dict(torch.load(hf_hub_download(repo_id="nickgardner/chatbot",
+                                      filename="alpaca_train_380_epoch.pt")))
 model.eval()
 def respond(custom_string):