Fix chat responses being cut off
This commit is contained in:
@@ -309,6 +309,7 @@ class Model:
|
|||||||
outputs = self.model.generate(
|
outputs = self.model.generate(
|
||||||
**inputs,
|
**inputs,
|
||||||
streamer=streamer,
|
streamer=streamer,
|
||||||
|
max_new_tokens=4096,
|
||||||
)
|
)
|
||||||
|
|
||||||
return self.tokenizer.decode(
|
return self.tokenizer.decode(
|
||||||
|
|||||||
Reference in New Issue
Block a user