theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-04-24 15:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

t2m

Browse Source
This commit is contained in:
mi804
2026-04-21 19:42:57 +08:00
parent 9d09e0431c
commit 95cfb77881
6 changed files with 135 additions and 261 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/diffusion/base_pipeline.py
View File

@@ -152,7 +152,7 @@ class BasePipeline(torch.nn.Module):
# remove batch dim
if audio_output.ndim == 3:
audio_output = audio_output.squeeze(0)
return audio_output.float()
return audio_output.float().cpu()
def load_models_to_device(self, model_names):
if self.vram_management_enabled: