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

fix typo in condirion
This commit is contained in:
Suraj Patil 2022-08-13 12:32:01 +05:30 committed by GitHub
parent c72e343085
commit 92b6dbba1a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions

View File

@ -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):

View File

@ -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):