lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-13 08:59:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

add sd2 template

Browse Source
This commit is contained in:
layerdiffusion
2024-08-05 11:49:45 -07:00
parent 27641043ee
commit 24cfce26dc
2 changed files with 83 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
backend/loader.py
View File

@@ -13,10 +13,11 @@ from backend.nn.clip import IntegratedCLIP, CLIPTextConfig
from backend.nn.unet import IntegratedUNet2DConditionModel
from backend.diffusion_engine.sd15 import StableDiffusion
from backend.diffusion_engine.sd20 import StableDiffusion2
from backend.diffusion_engine.sdxl import StableDiffusionXL
possible_models = [StableDiffusion, StableDiffusionXL]
possible_models = [StableDiffusion, StableDiffusion2, StableDiffusionXL]
logging.getLogger("diffusers").setLevel(logging.ERROR)