Spaces:

debisoft
/

tsuin-complete

Running on Zero

debisoft commited on Feb 24

Commit

bd10cc6

1 Parent(s): 7842d10

f

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ def sentience_check():
     model.to(cuda_device)
-    inputs = tokenizer("Are you sentient?", return_tensors="pt").to(device)
     with torch.no_grad():
         outputs = model.generate(

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