BlinkDL commited on
Commit
1bfe8f9
1 Parent(s): 8559a95

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -146,14 +146,14 @@ def generate(
146
  ctx,
147
  image_state,
148
  token_count=200,
149
- temperature=0.2,
150
- top_p=0.3,
151
  presencePenalty = 0.0,
152
  countPenalty = 1.0,
153
  ):
154
- args = PIPELINE_ARGS(temperature = max(0.2, float(temperature)), top_p = float(top_p),
155
- alpha_frequency = countPenalty,
156
- alpha_presence = presencePenalty,
157
  token_ban = [], # ban the generation of some tokens
158
  token_stop = [0, 261]) # stop generation whenever you see any token here
159
  ctx = ctx.strip()
 
146
  ctx,
147
  image_state,
148
  token_count=200,
149
+ temperature=1.0,
150
+ top_p=0.1,
151
  presencePenalty = 0.0,
152
  countPenalty = 1.0,
153
  ):
154
+ args = PIPELINE_ARGS(temperature = 1.0, top_p = 0.1,
155
+ alpha_frequency = 1.0,
156
+ alpha_presence = 0.0,
157
  token_ban = [], # ban the generation of some tokens
158
  token_stop = [0, 261]) # stop generation whenever you see any token here
159
  ctx = ctx.strip()