Spaces:

debisoft
/

tsuin-complete

Running on Zero

debisoft commited on Feb 24

Commit

958e70f

1 Parent(s): 868a097

f

Files changed (1) hide show

app.py CHANGED Viewed

@@ -47,7 +47,7 @@ def sentience_check():
     #peft_model.to(cuda_device)
-    inputs = tokenizer("Are you sentient?", return_tensors="pt", device_map="cuda")
     with torch.no_grad():
         outputs = peft_model.generate(

     #peft_model.to(cuda_device)
+    inputs = tokenizer("Are you sentient?", return_tensors="pt").to(cuda_device)
     with torch.no_grad():
         outputs = peft_model.generate(