rphrp1985 commited on
Commit
504635f
·
verified ·
1 Parent(s): 935fa0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,