Spaces:

selbl
/

LyricGeneration

Sleeping

selbl commited on Feb 28

Commit

3e231ed

•

1 Parent(s): 0b384a2

Update StreamlitModel.py

Files changed (1) hide show

StreamlitModel.py CHANGED Viewed

@@ -126,7 +126,7 @@ def TextGeneration(prompt,prof=False,parts=True):
     sample_outputs = generate(generated,
                              max_new_tokens=200,
-                             context_size=400,
                              tokenizer=tokenizer,
                              model=gpt_model,
                              top_k=10,

     sample_outputs = generate(generated,
                              max_new_tokens=200,
+                             context_size=450,
                              tokenizer=tokenizer,
                              model=gpt_model,
                              top_k=10,