Spaces:

broadfield
/

Song-Writer

Runtime error

broadfield commited on Dec 30, 2023

Commit

1bbc4ff

1 Parent(s): 25a21df

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,7 +20,7 @@ def generate(prompt,history):
         temperature=0.9,
         max_new_tokens=1024,
         top_p=0.95,
-        repetition_penalty=repetition_penalty,
         do_sample=True,
         seed=seed,
     )

         temperature=0.9,
         max_new_tokens=1024,
         top_p=0.95,
+        repetition_penalty=1.0,
         do_sample=True,
         seed=seed,
     )