Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ def chat(prompt, top_k, temperature):
|
|
19 |
generate_ids = model.generate(inputs.input_ids, max_new_tokens=2048, do_sample = True, top_k=int(top_k), top_p=0, temperature=float(temperature), repetition_penalty=1.15, eos_token_id=2, bos_token_id=1, pad_token_id=0)
|
20 |
response = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|
21 |
response = response.lstrip(prompt)
|
22 |
-
print('log:', response)
|
23 |
return response
|
24 |
|
25 |
|
|
|
19 |
generate_ids = model.generate(inputs.input_ids, max_new_tokens=2048, do_sample = True, top_k=int(top_k), top_p=0, temperature=float(temperature), repetition_penalty=1.15, eos_token_id=2, bos_token_id=1, pad_token_id=0)
|
20 |
response = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|
21 |
response = response.lstrip(prompt)
|
22 |
+
print('-log: ',prompt, response)
|
23 |
return response
|
24 |
|
25 |
|