Spaces:
Running
Running
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())
- 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(
|
|
|
|
|
|
|
|
|
|
| 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:
|