diff --git a/comfy/sd.py b/comfy/sd.py index ee350d5b5bd6..e98a3aa87cef 100644 --- a/comfy/sd.py +++ b/comfy/sd.py @@ -451,7 +451,7 @@ def __init__(self, sd=None, device=None, config=None, dtype=None, metadata=None) self.latent_dim = 2 self.process_output = lambda audio: audio self.process_input = lambda audio: audio - self.working_dtypes = [torch.bfloat16, torch.float32] + self.working_dtypes = [torch.bfloat16, torch.float16, torch.float32] self.disable_offload = True self.extra_1d_channel = 16 else: