[diffusion] chore: remove stepvideo code (#15918)
This commit is contained in:
@@ -752,8 +752,7 @@ class SchedulerLoader(ComponentLoader):
|
||||
scheduler = scheduler_cls(**config)
|
||||
if server_args.pipeline_config.flow_shift is not None:
|
||||
scheduler.set_shift(server_args.pipeline_config.flow_shift)
|
||||
if server_args.pipeline_config.timesteps_scale is not None:
|
||||
scheduler.set_timesteps_scale(server_args.pipeline_config.timesteps_scale)
|
||||
|
||||
return scheduler
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user