parent
c72e343085
commit
92b6dbba1a
|
@ -63,7 +63,7 @@ class LDMTextToImagePipeline(DiffusionPipeline):
|
|||
# prepare extra kwargs for the scheduler step, since not all schedulers have the same signature
|
||||
accepts_eta = "eta" in set(inspect.signature(self.scheduler.step).parameters.keys())
|
||||
extra_kwrags = {}
|
||||
if not accepts_eta:
|
||||
if accepts_eta:
|
||||
extra_kwrags["eta"] = eta
|
||||
|
||||
for t in tqdm(self.scheduler.timesteps):
|
||||
|
|
|
@ -36,7 +36,7 @@ class LDMPipeline(DiffusionPipeline):
|
|||
# prepare extra kwargs for the scheduler step, since not all schedulers have the same signature
|
||||
accepts_eta = "eta" in set(inspect.signature(self.scheduler.step).parameters.keys())
|
||||
extra_kwrags = {}
|
||||
if not accepts_eta:
|
||||
if accepts_eta:
|
||||
extra_kwrags["eta"] = eta
|
||||
|
||||
for t in tqdm(self.scheduler.timesteps):
|
||||
|
|
Loading…
Reference in New Issue