nazdridoy commited on
Commit
1675f59
·
verified ·
1 Parent(s): d8d1319

fix(image): pass model name explicitly for image-to-image

Browse files

- [fix] Explicitly pass `model` argument to `client.image_to_image()` (image_handler.py:generate_image_to_image():generate_image_task())

Files changed (1) hide show
  1. image_handler.py +4 -1
image_handler.py CHANGED
@@ -208,7 +208,10 @@ def generate_image_to_image(
208
 
209
  # Create generation function for timeout handling
210
  def generate_image_task():
211
- return client.image_to_image(**generation_params)
 
 
 
212
 
213
  # Execute with timeout using ThreadPoolExecutor
214
  with ThreadPoolExecutor(max_workers=1) as executor:
 
208
 
209
  # Create generation function for timeout handling
210
  def generate_image_task():
211
+ return client.image_to_image(
212
+ model=model_name,
213
+ **generation_params
214
+ )
215
 
216
  # Execute with timeout using ThreadPoolExecutor
217
  with ThreadPoolExecutor(max_workers=1) as executor: