Skip to content

Commit d9ad3f8

Browse files
committed
set default flow base schedule
1 parent 459a0cb commit d9ad3f8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/schedulers/schedules/flow_schedule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def __init__(
124124

125125
def set_base_schedule(self, base_schedule: Union[str]):
126126
if base_schedule is None:
127-
raise ValueError("Must set base schedule.")
127+
base_schedule = self.base_schedules["FlowMatchSD3"]
128128
if isinstance(base_schedule, str):
129129
if base_schedule not in self.base_schedules:
130130
raise ValueError(f"Expected one of {self.base_schedules.keys()}")

0 commit comments

Comments
 (0)