nschenone commited on
Commit
044d15c
1 Parent(s): 8987832

Updated generation default args

Browse files
Files changed (1) hide show
  1. src/generate.py +2 -2
src/generate.py CHANGED
@@ -13,7 +13,7 @@ def generate(
13
 
14
  set_seed(seed)
15
 
16
- default_pipline_args = {
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 = default_pipline_args.update(pipeline_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: