Commit d43333ff authored by AUTOMATIC1111's avatar AUTOMATIC1111
Browse files

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

parent 0cdbd90d

modules/sd_samplers_common.py

+2 −0
Original line number Diff line number Diff line
@@ -95,6 +95,8 @@ def images_tensor_to_samples(image, approximation=None, model=None): 
    else:

        if model is None:

            model = shared.sd_model

        model.first_stage_model.to(devices.dtype_vae)



        image = image.to(shared.device, dtype=devices.dtype_vae)

        image = image * 2 - 1

        if len(image) > 1:

CRA Git | Maintained and supported by SUSTech CRA and CCSE