lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-28 02:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

revise space

Browse Source
This commit is contained in:
layerdiffusion
2024-08-19 22:43:09 -07:00
parent fef6df29d9
commit 65ec461f8a
9 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions-builtin/forge_space_illusion_diffusion/forge_app.py
View File

@@ -28,7 +28,7 @@ from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionS
BASE_MODEL = "SG161222/Realistic_Vision_V5.1_noVAE"
with spaces.GPUObject() as gpu_object:
with spaces.capture_gpu_object() as gpu_object:
# Initialize both pipelines
vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse", torch_dtype=torch.float16)
controlnet = ControlNetModel.from_pretrained("monster-labs/control_v1p_sd15_qrcode_monster", torch_dtype=torch.float16)