Spaces:

howard-hou
/

VisualRWKV-Gradio-1

Sleeping

howard-hou commited on Dec 30, 2023

Commit

43e9368

•

1 Parent(s): b9bf03b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -41,8 +41,8 @@ def generate(
     token_count=200,
     temperature=0.2,
     top_p=0.3,
-    presencePenalty = 0.1,
-    countPenalty = 0.1,
 ):
     args = PIPELINE_ARGS(temperature = max(0.2, float(temperature)), top_p = float(top_p),
                     alpha_frequency = countPenalty,

     token_count=200,
     temperature=0.2,
     top_p=0.3,
+    presencePenalty = 0.0,
+    countPenalty = 1.0,
 ):
     args = PIPELINE_ARGS(temperature = max(0.2, float(temperature)), top_p = float(top_p),
                     alpha_frequency = countPenalty,