adding image param to refiner in code output
Browse files
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]'
|