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,