Spaces:

mikegarts
/

remarqify

Runtime error

mikegarts commited on Jun 23, 2022

Commit

2c32fbf

•

1 Parent(s): d9d4776

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def generate(prompt):
     model, tokenizer = get_model()
     input_context = prompt
-    input_ids = tokenizer.encode(input_context, return_tensors="pt").to('cuda')
     outputs = model.generate(
         input_ids=input_ids,

     model, tokenizer = get_model()
     input_context = prompt
+    input_ids = tokenizer.encode(input_context, return_tensors="pt").to('cpu')
     outputs = model.generate(
         input_ids=input_ids,