Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -33,12 +33,11 @@ def predict(message, history):
|
|
33 |
temperature=0.5,
|
34 |
repetition_penalty=1.2,
|
35 |
max_new_tokens=256,
|
36 |
-
streamer=streamer,
|
37 |
eos_token_id=tokenizer.eos_token_id,
|
38 |
pad_token_id=tokenizer.eos_token_id
|
39 |
)
|
40 |
|
41 |
-
outputs = model.generate(**inputs, generation_config=generation_config)
|
42 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
43 |
return response
|
44 |
|
|
|
33 |
temperature=0.5,
|
34 |
repetition_penalty=1.2,
|
35 |
max_new_tokens=256,
|
|
|
36 |
eos_token_id=tokenizer.eos_token_id,
|
37 |
pad_token_id=tokenizer.eos_token_id
|
38 |
)
|
39 |
|
40 |
+
outputs = model.generate(**inputs,streamer=streamer, generation_config=generation_config)
|
41 |
response = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
42 |
return response
|
43 |
|