AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

config fixes (#3060)

Browse Source
This commit is contained in:
Will Berman
2023-04-11 09:54:50 -07:00
committed by Daniel Gu
parent 1336668f0e
commit 2bc0a15818
4 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/community/sd_text2img_k_diffusion.py
View File

@@ -105,7 +105,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
)
model = ModelWrapper(unet, scheduler.alphas_cumprod)
if scheduler.prediction_type == "v_prediction":
if scheduler.config.prediction_type == "v_prediction":
self.k_diffusion_model = CompVisVDenoiser(model)
else:
self.k_diffusion_model = CompVisDenoiser(model)