nazdridoy commited on
Commit
d8d1319
·
verified ·
1 Parent(s): 43a0ca3

refactor(image): rename input image parameter key

Browse files

- [refactor] Change "input_image" key to "image" in generation_params (image_handler.py:generate_image_to_image():194)

Files changed (1) hide show
  1. image_handler.py +1 -1
image_handler.py CHANGED
@@ -192,7 +192,7 @@ def generate_image_to_image(
192
 
193
  # Prepare generation parameters
194
  generation_params = {
195
- "input_image": input_image,
196
  "prompt": prompt,
197
  "num_inference_steps": num_inference_steps,
198
  "guidance_scale": guidance_scale,
 
192
 
193
  # Prepare generation parameters
194
  generation_params = {
195
+ "image": input_image,
196
  "prompt": prompt,
197
  "num_inference_steps": num_inference_steps,
198
  "guidance_scale": guidance_scale,