jinymusim
/

dialogmodel

Text Generation

text-generation-inference

Model card Files Files and versions

jinymusim commited on Apr 6, 2023

Commit

6f6b1ce

·

1 Parent(s): 3983dbb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ tokenized_context = tokenizer.encode(user_utterance + tokenizer.eos_token, retur
 # generated a response, limit max_lenght to resonable size
 out_response = model.generate(tokenized_context,
-                                max_length=30,
                                 num_beams=2,
                                 no_repeat_ngram_size=2,
                                 early_stopping=True,

 # generated a response, limit max_lenght to resonable size
 out_response = model.generate(tokenized_context,
+                                max_length=100,
                                 num_beams=2,
                                 no_repeat_ngram_size=2,
                                 early_stopping=True,