Unverified Commit 92b6dbba authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[LDM pipeline] fix eta condition. (#171)

fix typo in condirion
parent c72e3430
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment