music_generation

Sleeping

IsraelRM commited on Mar 10

Commit

806fe1a

verified ·

1 Parent(s): cb26207

Update heartlib/pipelines/music_generation.py

Files changed (1) hide show

heartlib/pipelines/music_generation.py CHANGED Viewed

@@ -126,6 +126,7 @@ class HeartMuLaGenPipeline:
                 self.codec_path,
                 device_map=self.codec_device,
                 dtype=self.codec_dtype,
             )
         self.lazy_load = lazy_load

                 self.codec_path,
                 device_map=self.codec_device,
                 dtype=self.codec_dtype,
+                ignore_mismatched_sizes=True
             )
         self.lazy_load = lazy_load