Spaces:
Runtime error
Runtime error
Commit
·
cddb226
1
Parent(s):
2f23fb7
Update app.py
Browse files
app.py
CHANGED
@@ -65,7 +65,7 @@ def generateresponse(history, task):
|
|
65 |
PROMPT,
|
66 |
return_tensors="pt",
|
67 |
)
|
68 |
-
input_ids = inputs["input_ids"]
|
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,
|