Fix advanced scheduler behaviour in MultiDiffusionPipeline.

This commit is contained in:
Ryan Dick
2024-06-18 12:01:59 -04:00
parent 72be7e71e3
commit 99046a8145

View File

@ -95,7 +95,7 @@ class MultiDiffusionPipeline(StableDiffusionGeneratorPipeline):
# we are calling step() multiple times at the same timestep (once for each region batch), we must maintain a
# separate scheduler state for each region batch.
region_batch_schedulers: list[SchedulerMixin] = [
copy.copy(self.scheduler) for _ in multi_diffusion_conditioning
copy.deepcopy(self.scheduler) for _ in multi_diffusion_conditioning
]
callback(