Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -21,8 +21,8 @@ def process(message, history):
|
|
21 |
input_prompt = f"{input_prompt} ### Instruction:{message.strip()} ### Response:"
|
22 |
inputs = tokenizer(input_prompt, return_tensors="pt").to("cuda:0")
|
23 |
try:
|
24 |
-
generate_ids = model.generate(inputs.input_ids, max_new_tokens=2048, do_sample=True, top_k=
|
25 |
-
temperature=
|
26 |
pad_token_id=0)
|
27 |
response = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|
28 |
print('log:', response)
|
|
|
21 |
input_prompt = f"{input_prompt} ### Instruction:{message.strip()} ### Response:"
|
22 |
inputs = tokenizer(input_prompt, return_tensors="pt").to("cuda:0")
|
23 |
try:
|
24 |
+
generate_ids = model.generate(inputs.input_ids, max_new_tokens=2048, do_sample=True, top_k=20, top_p=0.84,
|
25 |
+
temperature=1, repetition_penalty=1.15, eos_token_id=2, bos_token_id=1,
|
26 |
pad_token_id=0)
|
27 |
response = tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
|
28 |
print('log:', response)
|