Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion.py
View File

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

2
src/diffusers/pipelines/latent_diffusion_uncond/pipeline_latent_diffusion_uncond.py
View File

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