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

Fix LivePreview not working with fp8 UNet (#2412)

Browse Source 
* fix dtype
This commit is contained in:
Haoming
2024-12-05 19:33:35 +08:00
committed by GitHub
parent bae1bba891
commit 9fbba69297
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/devices.py
View File

@@ -54,7 +54,7 @@ device_interrogate: torch.device = memory_management.text_encoder_device() # fo
device_gfpgan: torch.device = memory_management.get_torch_device() # will be managed by memory management system device_gfpgan: torch.device = memory_management.get_torch_device() # will be managed by memory management system
device_esrgan: torch.device = memory_management.get_torch_device() # will be managed by memory management system device_esrgan: torch.device = memory_management.get_torch_device() # will be managed by memory management system
device_codeformer: torch.device = memory_management.get_torch_device() # will be managed by memory management system device_codeformer: torch.device = memory_management.get_torch_device() # will be managed by memory management system
dtype: torch.dtype = memory_management.unet_dtype() dtype: torch.dtype = torch.float32 if memory_management.unet_dtype() is torch.float32 else torch.float16
dtype_vae: torch.dtype = memory_management.vae_dtype() dtype_vae: torch.dtype = memory_management.vae_dtype()
dtype_unet: torch.dtype = memory_management.unet_dtype() dtype_unet: torch.dtype = memory_management.unet_dtype()
dtype_inference: torch.dtype = memory_management.unet_dtype() dtype_inference: torch.dtype = memory_management.unet_dtype()