GhadeerALbadani commited on
Commit
7646c36
·
1 Parent(s): 3125d16

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -6
README.md CHANGED
@@ -39,11 +39,7 @@ generation_pipeline = pipeline("text-generation",model=model,tokenizer=tokenizer
39
  #feel free to try different decoding settings
40
  generation_pipeline(text,
41
  pad_token_id=tokenizer.eos_token_id,
42
- num_beams=10,
43
- max_length=512,
44
  penalty_alpha=0.6,
45
- top_p=0.9,
46
- top_k=4,
47
- repetition_penalty = 3.0,
48
- no_repeat_ngram_size = 3)[0]['generated_text']
49
  ```
 
39
  #feel free to try different decoding settings
40
  generation_pipeline(text,
41
  pad_token_id=tokenizer.eos_token_id,
42
+ max_length=512,
 
43
  penalty_alpha=0.6,
44
+ top_k=4 )[0]['generated_text']
 
 
 
45
  ```