Update README.md
Browse files
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(' ', '')
|