From 3adac330c2b58dfd3f741b8772d20c40b3212e45 Mon Sep 17 00:00:00 2001 From: ciaranbor Date: Mon, 5 Jan 2026 11:15:35 +0000 Subject: [PATCH] Use 10% sync steps --- src/exo/worker/engines/image/config.py | 2 +- src/exo/worker/engines/image/models/flux/config.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/exo/worker/engines/image/config.py b/src/exo/worker/engines/image/config.py index a1086b9f..93bb8369 100644 --- a/src/exo/worker/engines/image/config.py +++ b/src/exo/worker/engines/image/config.py @@ -38,7 +38,7 @@ class ImageModelConfig(BaseModel): # Inference parameters default_steps: dict[str, int] # {"low": X, "medium": Y, "high": Z} - num_sync_steps_factor: float # Fraction of medium steps for sync phase + num_sync_steps_factor: float # Fraction of steps for sync phase # Feature flags uses_attention_mask: bool # True for Fibo diff --git a/src/exo/worker/engines/image/models/flux/config.py b/src/exo/worker/engines/image/models/flux/config.py index 6d03bd1f..df30c5c0 100644 --- a/src/exo/worker/engines/image/models/flux/config.py +++ b/src/exo/worker/engines/image/models/flux/config.py @@ -43,6 +43,6 @@ FLUX_DEV_CONFIG = ImageModelConfig( patch_size=2, vae_scale_factor=8, default_steps={"low": 10, "medium": 25, "high": 50}, - num_sync_steps_factor=0.125, # ~3 sync steps for medium (25 steps) + num_sync_steps_factor=0.1, # ~3 sync steps for medium (25 steps) uses_attention_mask=False, )