Commit
·
cce57a3
1
Parent(s):
78053ef
Update README.md
Browse files
README.md
CHANGED
@@ -34,7 +34,7 @@ happy_gen = HappyGeneration("GPT-NEO", "DarwinAnim8or/gpt-grug-125m")
|
|
34 |
|
35 |
#Set generation settings:
|
36 |
from happytransformer import GENSettings
|
37 |
-
args_top_k = GENSettings(no_repeat_ngram_size=3, do_sample=True,top_k=50, temperature=0.7, max_length=50, early_stopping=False)
|
38 |
|
39 |
#Generate a response:
|
40 |
result = happy_gen.generate_text("""Person: "Hello grug"
|
|
|
34 |
|
35 |
#Set generation settings:
|
36 |
from happytransformer import GENSettings
|
37 |
+
args_top_k = GENSettings(no_repeat_ngram_size=3, do_sample=True,top_k=50, temperature=0.7, max_length=50, early_stopping=False)
|
38 |
|
39 |
#Generate a response:
|
40 |
result = happy_gen.generate_text("""Person: "Hello grug"
|