minko186 commited on
Commit
3076727
·
verified ·
1 Parent(s): 9b0f802

Update gptzero_free.py

Browse files
Files changed (1) hide show
  1. gptzero_free.py +1 -1
gptzero_free.py CHANGED
@@ -14,7 +14,7 @@ from collections import OrderedDict
14
 
15
 
16
  class GPT2PPL:
17
- def __init__(self, device="cuda", model_id="gpt2"):
18
  self.device = device
19
  self.model_id = model_id
20
  self.model = GPT2LMHeadModel.from_pretrained(model_id).to(device)
 
14
 
15
 
16
  class GPT2PPL:
17
+ def __init__(self, device="cpu", model_id="gpt2"):
18
  self.device = device
19
  self.model_id = model_id
20
  self.model = GPT2LMHeadModel.from_pretrained(model_id).to(device)