Spaces:
Build error
Build error
Updated generation default args
Browse files- src/generate.py +2 -2
src/generate.py
CHANGED
@@ -13,7 +13,7 @@ def generate(
|
|
13 |
|
14 |
set_seed(seed)
|
15 |
|
16 |
-
|
17 |
"text_inputs": None,
|
18 |
"max_length": 100,
|
19 |
"num_return_sequences": 1,
|
@@ -24,7 +24,7 @@ def generate(
|
|
24 |
"temperature": 1.5,
|
25 |
}
|
26 |
|
27 |
-
args
|
28 |
generated = pipeline(**args)[0]["generated_text"]
|
29 |
|
30 |
if censor:
|
|
|
13 |
|
14 |
set_seed(seed)
|
15 |
|
16 |
+
args = {
|
17 |
"text_inputs": None,
|
18 |
"max_length": 100,
|
19 |
"num_return_sequences": 1,
|
|
|
24 |
"temperature": 1.5,
|
25 |
}
|
26 |
|
27 |
+
args.update(pipeline_args)
|
28 |
generated = pipeline(**args)[0]["generated_text"]
|
29 |
|
30 |
if censor:
|