jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

ACE VAE works in fp16. (#8055)

Browse Source
This commit is contained in:
comfyanonymous
2025-05-11 01:58:00 -07:00
committed by GitHub
parent 3535909eb8
commit 577de83ca9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd.py
View File

@@ -451,7 +451,7 @@ class VAE:
self.latent_dim = 2 self.latent_dim = 2
self.process_output = lambda audio: audio self.process_output = lambda audio: audio
self.process_input = 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.disable_offload = True
self.extra_1d_channel = 16 self.extra_1d_channel = 16
else: else: