Skip to content

Commit 69ac5ac

Browse files
Tuan TranTuan Tran
authored andcommitted
fix python 3.11 compatibility
1 parent a0cdecd commit 69ac5ac

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

lcm/train/two_tower_diffusion_lcm/criterion.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,9 @@ class TowerDiffusionLCMCriterionConfig(LCMCriterionConfig):
3333
Note that this requires the model to be set with
3434
`trained_with_cf_guidance = True`!
3535
"""
36-
step_sampling: StepsSamplerConfig = field(default_factory=lambda: StepsSamplerConfig())
36+
step_sampling: StepsSamplerConfig = field(
37+
default_factory=lambda: StepsSamplerConfig()
38+
)
3739

3840
log_losses_per_timestep_bucket: bool = False
3941

0 commit comments

Comments
 (0)