Loading modules/sd_samplers_common.py +2 −1 Original line number Diff line number Diff line Loading @@ -49,6 +49,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None): else: if model is None: model = shared.sd_model with devices.without_autocast(): # fixes an issue with unstable VAEs that are flaky even in fp32 x_sample = model.decode_first_stage(sample.to(model.first_stage_model.dtype)) return x_sample Loading Loading
modules/sd_samplers_common.py +2 −1 Original line number Diff line number Diff line Loading @@ -49,6 +49,7 @@ def samples_to_images_tensor(sample, approximation=None, model=None): else: if model is None: model = shared.sd_model with devices.without_autocast(): # fixes an issue with unstable VAEs that are flaky even in fp32 x_sample = model.decode_first_stage(sample.to(model.first_stage_model.dtype)) return x_sample Loading
