zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 18:10:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implement Minimax Music 3 + Core Support for Cuda Graphs (#15570)

Browse Source
This commit is contained in:
rattus
2026-08-14 02:10:08 +10:00
committed by GitHub
parent 12666983cb
commit efd4e951a0
19 changed files with 1333 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
comfy/model_base.py
View File

@@ -22,6 +22,7 @@ import torch
import logging
import comfy.ldm.lightricks.av_model
import comfy.ldm.minimax.model
import comfy.ldm.minimax_music.dit
import comfy.nested_tensor
import comfy.ldm.lightricks.symmetric_patchifier
import comfy.context_windows
@@ -2337,6 +2338,18 @@ class ACEStep15(BaseModel):
out['refer_audio'] = comfy.conds.CONDRegular(refer_audio)
return out
class MiniMaxMusic3(BaseModel):
def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.minimax_music.dit.MiniMaxMusic3DiT)
def process_timestep(self, timestep, **kwargs):
return 1.0 - timestep
def extra_conds(self, **kwargs):
out = super().extra_conds(**kwargs)
out["conditioning_scale"] = comfy.conds.CONDRegular(kwargs["conditioning_scale"])
return out
class Omnigen2(BaseModel):
def __init__(self, model_config, model_type=ModelType.FLOW, device=None):
super().__init__(model_config, model_type, device=device, unet_model=comfy.ldm.omnigen.omnigen2.OmniGen2Transformer2DModel)