mirrors/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-09-13 04:55:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make japanese hiragana and katakana characters work with ACE. (#7997)

Browse Source
This commit is contained in:
comfyanonymous
2025-05-08 00:32:36 -07:00
committed by GitHub
parent c7c025b8d1
commit 5d3cc85e13
3 changed files with 135 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/sd.py
View File

@@ -441,7 +441,7 @@ class VAE:
self.working_dtypes = [torch.float16, torch.bfloat16, torch.float32]
elif "vocoder.backbone.channel_layers.0.0.bias" in sd: #Ace Step Audio
self.first_stage_model = comfy.ldm.ace.vae.music_dcae_pipeline.MusicDCAE(source_sample_rate=44100)
self.memory_used_encode = lambda shape, dtype: (shape[2] * 300) * model_management.dtype_size(dtype)
self.memory_used_encode = lambda shape, dtype: (shape[2] * 330) * model_management.dtype_size(dtype)
self.memory_used_decode = lambda shape, dtype: (shape[2] * shape[3] * 87000) * model_management.dtype_size(dtype)
self.latent_channels = 8
self.output_channels = 2