Update BlackForestLabs_Flux1KontextDev.py

This commit is contained in:
H Lohaus 2025-07-11 09:25:14 +02:00 committed by GitHub
parent f8f4a21faf
commit 56948ae387
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -137,7 +137,7 @@ class BlackForestLabs_Flux1KontextDev(AsyncGeneratorProvider, ProviderModelMixin
elif chunk.get("msg") == "progress":
progress_data = chunk.get("progress_data", [])
progress_data = progress_data[0] if progress_data else {}
yield Reasoning(label="Processing image", status=f"{progress_data.get("index", 0)}/{progress_data.get("length", 0)}")
yield Reasoning(label="Processing image", status=f"{progress_data.get('index', 0)}/{progress_data.get('length', 0)}")
elif chunk.get("msg") == "process_completed":
url = chunk.get("output", {}).get("data", [{}])[0].get("url")
yield ImageResponse(url, prompt)