Fix advanced scheduler behaviour in MultiDiffusionPipeline.

This commit is contained in:
Ryan Dick 2024-06-18 12:01:59 -04:00 committed by Kent Keirsey
parent c22526b9d0
commit fb0aaa3e6d

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 # 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. # separate scheduler state for each region batch.
region_batch_schedulers: list[SchedulerMixin] = [ region_batch_schedulers: list[SchedulerMixin] = [
copy.copy(self.scheduler) for _ in multi_diffusion_conditioning copy.deepcopy(self.scheduler) for _ in multi_diffusion_conditioning
] ]
callback( callback(