jed351 commited on
Commit
a2171f4
1 Parent(s): 7dcaf95

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -4
README.md CHANGED
@@ -77,10 +77,7 @@ model = AutoModelForCausalLM.from_pretrained("jed351/gpt2_tiny_zh-hk-shikoto")
77
  # try messing around with the parameters
78
  generator = TextGenerationPipeline(model, tokenizer,
79
  max_new_tokens=200,
80
- no_repeat_ngram_size=3,
81
- do_sample=True,
82
- top_k=50, top_p=0.95,
83
- temperature=0.9) #, device=0) #if you have a GPU
84
 
85
  input_string = "your input"
86
 
 
77
  # try messing around with the parameters
78
  generator = TextGenerationPipeline(model, tokenizer,
79
  max_new_tokens=200,
80
+ no_repeat_ngram_size=3) #, device=0) #if you have a GPU
 
 
 
81
 
82
  input_string = "your input"
83