svjack
/

bloom-dialogue

Text2Text Generation

text-generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

svjack commited on Feb 20, 2023

Commit

95345c3

•

1 Parent(s): 316664a

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -16,7 +16,7 @@ tokenizer.decode(
 model.generate(
     tokenizer.encode(
             "你认为今天的天气怎么样？", return_tensors="pt", add_special_tokens=True
-        ), 128,
     num_beams=2,
     top_p = 0.95,
     top_k = 50,

 model.generate(
     tokenizer.encode(
             "你认为今天的天气怎么样？", return_tensors="pt", add_special_tokens=True
+        ), max_length = 128,
     num_beams=2,
     top_p = 0.95,
     top_k = 50,