AUTOMATIC1111/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-03-05 21:20:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve efficiency and support more device

Browse Source
This commit is contained in:
Kohaku-Blueleaf
2024-01-09 22:11:44 +08:00
parent 6869d95890
commit 209c26a1cb
2 changed files with 44 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modules/shared_init.py
View File

@@ -29,6 +29,7 @@ def initialize():
devices.dtype = torch.float32 if cmd_opts.no_half else torch.float16
devices.dtype_vae = torch.float32 if cmd_opts.no_half or cmd_opts.no_half_vae else torch.float16
devices.dtype_inference = torch.float32 if cmd_opts.precision == 'full' else devices.dtype
shared.device = devices.device
shared.weight_load_location = None if cmd_opts.lowram else "cpu"