Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -111,7 +111,7 @@ def engine(input_embeds):
|
|
111 |
batch_input_embeds = input_embeds[start:start+batch_size]
|
112 |
outputs = m.generate(
|
113 |
inputs_embeds = batch_input_embeds.to("cuda"),
|
114 |
-
max_new_tokens =
|
115 |
do_sample = True,
|
116 |
temperature = 0.6,
|
117 |
top_p = 0.9,
|
|
|
111 |
batch_input_embeds = input_embeds[start:start+batch_size]
|
112 |
outputs = m.generate(
|
113 |
inputs_embeds = batch_input_embeds.to("cuda"),
|
114 |
+
max_new_tokens = 512,
|
115 |
do_sample = True,
|
116 |
temperature = 0.6,
|
117 |
top_p = 0.9,
|