mirror/exo
1
0
Fork 0
mirror of https://github.com/exo-explore/exo.git synced 2026-01-21 20:39:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use 10% sync steps

Browse Source
This commit is contained in:
ciaranbor
2026-01-05 11:15:35 +00:00
parent 33507c950b
commit 3adac330c2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/exo/worker/engines/image/config.py
View File

@@ -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

2
src/exo/worker/engines/image/models/flux/config.py
View File

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