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, )