Spaces:

JosephH
/

Test

Runtime error

JosephH commited on Mar 23, 2023

Commit

92ddfde

1 Parent(s): 09112cf

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -14,7 +14,7 @@ from collections import OrderedDict
 class GPT2PPL:
-    def __init__(self, device="cuda", model_id="gpt2"):
         self.device = device
         self.model_id = model_id
         self.model = GPT2LMHeadModel.from_pretrained(model_id).to(device)

 class GPT2PPL:
+    def __init__(self, device="cpu", model_id="gpt2"):
         self.device = device
         self.model_id = model_id
         self.model = GPT2LMHeadModel.from_pretrained(model_id).to(device)