n42 commited on
Commit
a5752aa
·
1 Parent(s): c6a81f6

adding image param to refiner in code output

Browse files
Files changed (1) hide show
  1. config.py +2 -1
config.py CHANGED
@@ -182,7 +182,8 @@ def assemble_code(str_config):
182
  code["110_run_refiner"] = f'''image = refiner(
183
  prompt=prompt,
184
  negative_prompt=negative_prompt,
185
- num_inference_steps=inference_steps
 
186
  ).images[0]'''
187
 
188
  code["200_show_image"] = 'image[0]'
 
182
  code["110_run_refiner"] = f'''image = refiner(
183
  prompt=prompt,
184
  negative_prompt=negative_prompt,
185
+ num_inference_steps=inference_steps,
186
+ image=image
187
  ).images[0]'''
188
 
189
  code["200_show_image"] = 'image[0]'