Update README.md
Browse files
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 |
|