mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-11-21 03:11:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix an issue where VAE would remain in fp16 after an auto-switch to fp32

Browse Source
This commit is contained in:
AUTOMATIC1111 2023-08-30 21:13:24 +03:00
parent 0cdbd90d6b
commit d43333ff71
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/sd_samplers_common.py
View File

@ -95,6 +95,8 @@ def images_tensor_to_samples(image, approximation=None, model=None):
else: else:
if model is None: if model is None:
model = shared.sd_model model = shared.sd_model
model.first_stage_model.to(devices.dtype_vae)
image = image.to(shared.device, dtype=devices.dtype_vae) image = image.to(shared.device, dtype=devices.dtype_vae)
image = image * 2 - 1 image = image * 2 - 1
if len(image) > 1: if len(image) > 1: