jed351 commited on
Commit
5521afc
·
1 Parent(s): b615d66

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -4
README.md CHANGED
@@ -48,10 +48,7 @@ model = AutoModelForCausalLM.from_pretrained("jed351/gpt2_base_zh-hk-shikoto")
48
  # try messing around with the parameters
49
  generator = TextGenerationPipeline(model, tokenizer,
50
  max_new_tokens=200,
51
- no_repeat_ngram_size=3,
52
- do_sample=True,
53
- top_k=50, top_p=0.95,
54
- temperature=0.9) #, device=0) #if you have a GPU
55
  input_string = "your input"
56
  output = generator(input_string)
57
  string = output[0]['generated_text'].replace(' ', '')
 
48
  # try messing around with the parameters
49
  generator = TextGenerationPipeline(model, tokenizer,
50
  max_new_tokens=200,
51
+ no_repeat_ngram_size=3) #, device=0) #if you have a GPU
 
 
 
52
  input_string = "your input"
53
  output = generator(input_string)
54
  string = output[0]['generated_text'].replace(' ', '')