Spaces:

zachlopez
/

sample_1

Runtime error

zachlopez commited on Apr 2, 2022

Commit

5b1c0e6

1 Parent(s): 795dec4

moved to device

Files changed (1) hide show

app.py CHANGED Viewed

@@ -800,7 +800,7 @@ model = GPT2LMHeadModel.from_pretrained(
     output_hidden_states=True
 )
 if model_fp != None and model_fp != "":
-    model.load_state_dict(torch.load(model_fp))
 model.to(device)
 model.eval()

     output_hidden_states=True
 )
 if model_fp != None and model_fp != "":
+    model.load_state_dict(torch.load(model_fp, map_location=device))
 model.to(device)
 model.eval()