Commit 9a749ea4 authored by Your Name's avatar Your Name

Fix diffusers time variable scoping issue

- Import time module inside try block with alias to avoid UnboundLocalError
- This prevents Python's exception handling from affecting variable scope
parent 9f01de41
...@@ -3766,6 +3766,10 @@ async def create_image_generation(request: ImageGenerationRequest, http_request: ...@@ -3766,6 +3766,10 @@ async def create_image_generation(request: ImageGenerationRequest, http_request:
multi_model_manager.add_model(model_key, pipeline) multi_model_manager.add_model(model_key, pipeline)
# Get timestamp BEFORE calling diffusers (to avoid scope issues)
import time as time_module
timestamp = int(time_module.time())
# Generate images # Generate images
# Use request seed if provided, otherwise use CLI default seed # Use request seed if provided, otherwise use CLI default seed
seed = request.seed if request.seed is not None else getattr(global_args, 'image_seed', None) seed = request.seed if request.seed is not None else getattr(global_args, 'image_seed', None)
...@@ -3809,7 +3813,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request: ...@@ -3809,7 +3813,7 @@ async def create_image_generation(request: ImageGenerationRequest, http_request:
images.append(img_data) images.append(img_data)
return { return {
"created": int(time.time()), "created": timestamp,
"data": images "data": images
} }
......
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