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): ...@@ -63,7 +63,7 @@ class LDMTextToImagePipeline(DiffusionPipeline):
# prepare extra kwargs for the scheduler step, since not all schedulers have the same signature # 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()) accepts_eta = "eta" in set(inspect.signature(self.scheduler.step).parameters.keys())
extra_kwrags = {} extra_kwrags = {}
if not accepts_eta: if accepts_eta:
extra_kwrags["eta"] = eta extra_kwrags["eta"] = eta
for t in tqdm(self.scheduler.timesteps): for t in tqdm(self.scheduler.timesteps):
......
...@@ -36,7 +36,7 @@ class LDMPipeline(DiffusionPipeline): ...@@ -36,7 +36,7 @@ class LDMPipeline(DiffusionPipeline):
# prepare extra kwargs for the scheduler step, since not all schedulers have the same signature # 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()) accepts_eta = "eta" in set(inspect.signature(self.scheduler.step).parameters.keys())
extra_kwrags = {} extra_kwrags = {}
if not accepts_eta: if accepts_eta:
extra_kwrags["eta"] = eta extra_kwrags["eta"] = eta
for t in tqdm(self.scheduler.timesteps): 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