Spaces:

anirudh-sub
/

Debate_v2

Runtime error

anirudh-sub commited on Oct 1, 2023

Commit

6d71a04

1 Parent(s): d68a791

Debugged app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ pipeline = transformers.pipeline(
     device_map="auto",
 )
-torch.set_default_tensor_type(torch.cuda.Tensor)
 def debate_response(text):
     sequences = pipeline(

     device_map="auto",
 )
+torch.set_default_tensor_type(torch.FloatTensor)
 def debate_response(text):
     sequences = pipeline(