lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-14 01:19:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

intergrate k-diffusion

Browse Source
This commit is contained in:
lllyasviel
2024-08-07 15:05:42 -07:00
committed by GitHub
parent 14a759b5ca
commit a07c758658
14 changed files with 1366 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/diffusion_engine/base.py
View File

@@ -39,8 +39,6 @@ class ForgeDiffusionEngine:
self.is_sd2 = False
self.is_sdxl = False
self.is_sd3 = False
self.parameterization = 'eps'
self.alphas_cumprod = None
def set_clip_skip(self, clip_skip):
pass

1
backend/diffusion_engine/sd15.py
View File

@@ -53,7 +53,6 @@ class StableDiffusion(ForgeDiffusionEngine):
# WebUI Legacy
self.is_sd1 = True
self.first_stage_model = vae.first_stage_model
self.alphas_cumprod = unet.model.predictor.alphas_cumprod
def set_clip_skip(self, clip_skip):
self.text_processing_engine.clip_skip = clip_skip

4
backend/diffusion_engine/sd20.py
View File

@@ -53,10 +53,6 @@ class StableDiffusion2(ForgeDiffusionEngine):
# WebUI Legacy
self.is_sd2 = True
self.first_stage_model = vae.first_stage_model
self.alphas_cumprod = unet.model.predictor.alphas_cumprod
if not self.is_inpaint:
self.parameterization = 'v'
def set_clip_skip(self, clip_skip):
self.text_processing_engine.clip_skip = clip_skip

1
backend/diffusion_engine/sdxl.py
View File

@@ -72,7 +72,6 @@ class StableDiffusionXL(ForgeDiffusionEngine):
# WebUI Legacy
self.is_sdxl = True
self.first_stage_model = vae.first_stage_model
self.alphas_cumprod = unet.model.predictor.alphas_cumprod
def set_clip_skip(self, clip_skip):
self.text_processing_engine_l.clip_skip = clip_skip