Update app.py
Browse files
app.py
CHANGED
@@ -38,8 +38,7 @@ def load_pipelines():
|
|
38 |
top_p=0.9,
|
39 |
repetition_penalty=1.1,
|
40 |
no_repeat_ngram_size=2,
|
41 |
-
max_new_tokens=120
|
42 |
-
return_full_text=False
|
43 |
)
|
44 |
return captioner, storyteller
|
45 |
|
|
|
38 |
top_p=0.9,
|
39 |
repetition_penalty=1.1,
|
40 |
no_repeat_ngram_size=2,
|
41 |
+
max_new_tokens=120
|
|
|
42 |
)
|
43 |
return captioner, storyteller
|
44 |
|