jinymusim
/

gpt-czech-poet-our

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

jinymusim commited on Apr 29

Commit

06abe21

•

1 Parent(s): 286b230

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -39,8 +39,8 @@ tokenized_poet_start = tokenizer.encode(poet_start, return_tensors='pt')
 # generated a continuation to it
 out = model.generate(tokenized_poet_start,
                                 max_length=256,
-                                num_beams=8,
-                                no_repeat_ngram_size=2,
                                 early_stopping=True,
                                 pad_token_id= tokenizer.pad_token_id,
                                 eos_token_id = tokenizer.eos_token_id)

 # generated a continuation to it
 out = model.generate(tokenized_poet_start,
                                 max_length=256,
+                                do_sample=True,
+                                top_k=50
                                 early_stopping=True,
                                 pad_token_id= tokenizer.pad_token_id,
                                 eos_token_id = tokenizer.eos_token_id)