Commit 3be02b8c authored by Your Name's avatar Your Name

Fix bugs in image generation endpoint

- Fix missing indentation in async with semaphore block
- Fix invalid elif syntax in load_mode determination
- Fix request.steps reference (field doesn't exist in request model)
parent 2c0503d9
...@@ -3175,7 +3175,7 @@ async def create_image_generation(request: ImageGenerationRequest): ...@@ -3175,7 +3175,7 @@ async def create_image_generation(request: ImageGenerationRequest):
) )
async with semaphore: async with semaphore:
image_model = multi_model_manager.image_model image_model = multi_model_manager.image_model
# If no image model configured, return an error # If no image model configured, return an error
if not image_model: if not image_model:
...@@ -3224,8 +3224,8 @@ async def create_image_generation(request: ImageGenerationRequest): ...@@ -3224,8 +3224,8 @@ async def create_image_generation(request: ImageGenerationRequest):
except ValueError: except ValueError:
pass pass
# Determine number of steps # Use default steps for Z-Image Turbo (very fast)
steps = request.steps if request.steps else 4 steps = 4 # Default for fast generation
# Generate images using sd.cpp (run in thread to not block event loop) # Generate images using sd.cpp (run in thread to not block event loop)
result = await asyncio.to_thread( result = await asyncio.to_thread(
...@@ -4622,11 +4622,12 @@ def main(): ...@@ -4622,11 +4622,12 @@ def main():
load_mode = "ondemand" load_mode = "ondemand"
if args.loadall: if args.loadall:
load_mode = "loadall" load_mode = "loadall"
# Update global load_mode for semaphore system
load_mode["mode"] = load_mode
elif args.loadswap: elif args.loadswap:
load_mode = "loadswap" load_mode = "loadswap"
# Update global load_mode for semaphore system
load_mode_dict = {"mode": load_mode}
# Set load mode in multi_model_manager # Set load mode in multi_model_manager
multi_model_manager.set_load_mode(load_mode) multi_model_manager.set_load_mode(load_mode)
...@@ -4693,11 +4694,12 @@ def main(): ...@@ -4693,11 +4694,12 @@ def main():
load_mode = "ondemand" load_mode = "ondemand"
if args.loadall: if args.loadall:
load_mode = "loadall" load_mode = "loadall"
# Update global load_mode for semaphore system
load_mode["mode"] = load_mode
elif args.loadswap: elif args.loadswap:
load_mode = "loadswap" load_mode = "loadswap"
# Update global load_mode for semaphore system
load_mode_dict = {"mode": load_mode}
# Set load mode in multi_model_manager # Set load mode in multi_model_manager
multi_model_manager.set_load_mode(load_mode) multi_model_manager.set_load_mode(load_mode)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment