Update app.py
Browse files
app.py
CHANGED
@@ -60,7 +60,7 @@ def respond(
|
|
60 |
## <BOS_TOKEN><|START_OF_TURN_TOKEN|><|USER_TOKEN|>Hello, how are you?<|END_OF_TURN_TOKEN|><|START_OF_TURN_TOKEN|><|CHATBOT_TOKEN|>
|
61 |
|
62 |
gen_tokens = model.generate(
|
63 |
-
input_ids,
|
64 |
max_new_tokens=100,
|
65 |
do_sample=True,
|
66 |
temperature=0.3,
|
|
|
60 |
## <BOS_TOKEN><|START_OF_TURN_TOKEN|><|USER_TOKEN|>Hello, how are you?<|END_OF_TURN_TOKEN|><|START_OF_TURN_TOKEN|><|CHATBOT_TOKEN|>
|
61 |
|
62 |
gen_tokens = model.generate(
|
63 |
+
input_ids.float(),
|
64 |
max_new_tokens=100,
|
65 |
do_sample=True,
|
66 |
temperature=0.3,
|