mrleo1nid/ACE-Step
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add torch_dtype when load to avoice oom for windows

Browse Source
This commit is contained in:
chuxij
2025-05-05 04:51:58 +00:00
parent dce44fed36
commit bc84acb574
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pipeline_ace_step.py
+1 -1
View File
@@ -143,7 +143,7 @@ class ACEStepPipeline:
self.music_dcae = MusicDCAE(dcae_checkpoint_path=dcae_checkpoint_path, vocoder_checkpoint_path=vocoder_checkpoint_path) self.music_dcae = MusicDCAE(dcae_checkpoint_path=dcae_checkpoint_path, vocoder_checkpoint_path=vocoder_checkpoint_path)
self.music_dcae.to(device).eval().to(self.dtype) self.music_dcae.to(device).eval().to(self.dtype)
self.ace_step_transformer = ACEStepTransformer2DModel.from_pretrained(ace_step_checkpoint_path) self.ace_step_transformer = ACEStepTransformer2DModel.from_pretrained(ace_step_checkpoint_path, torch_dtype=self.dtype)
self.ace_step_transformer.to(device).eval().to(self.dtype) self.ace_step_transformer.to(device).eval().to(self.dtype)
lang_segment = LangSegment() lang_segment = LangSegment()