Spaces:
Running
on
T4
Running
on
T4
Upload app.py
Browse files
app.py
CHANGED
@@ -146,14 +146,14 @@ def generate(
|
|
146 |
ctx,
|
147 |
image_state,
|
148 |
token_count=200,
|
149 |
-
temperature=0
|
150 |
-
top_p=0.
|
151 |
presencePenalty = 0.0,
|
152 |
countPenalty = 1.0,
|
153 |
):
|
154 |
-
args = PIPELINE_ARGS(temperature =
|
155 |
-
alpha_frequency =
|
156 |
-
alpha_presence =
|
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()
|