thepolymerguy commited on
Commit
cddb226
·
1 Parent(s): 2f23fb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def generateresponse(history, task):
65
  PROMPT,
66
  return_tensors="pt",
67
  )
68
- input_ids = inputs["input_ids"].cuda()
69
 
70
  generation_config = GenerationConfig(
71
  temperature=0.6,
 
65
  PROMPT,
66
  return_tensors="pt",
67
  )
68
+ input_ids = inputs["input_ids"]#.cuda()
69
 
70
  generation_config = GenerationConfig(
71
  temperature=0.6,