Skip to content

Commit 577de83

Browse files
ACE VAE works in fp16. (#8055)
1 parent 3535909 commit 577de83

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/sd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -451,7 +451,7 @@ def __init__(self, sd=None, device=None, config=None, dtype=None, metadata=None)
451451
self.latent_dim = 2
452452
self.process_output = lambda audio: audio
453453
self.process_input = lambda audio: audio
454-
self.working_dtypes = [torch.bfloat16, torch.float32]
454+
self.working_dtypes = [torch.bfloat16, torch.float16, torch.float32]
455455
self.disable_offload = True
456456
self.extra_1d_channel = 16
457457
else:

0 commit comments

Comments
 (0)