Federico Volovnik-Jose
284d50b44c
- Use relative imports across nodes/ and lib/ so ComfyUI's loader resolves the local subpackages instead of colliding with ComfyUI's own top-level nodes module. - tts_output_to_audio_dict now returns CPU tensors so downstream nodes don't see CUDA tensors. - model_state cache key uses resolved dtype string, so dtype=auto and the matching explicit choice share one cache slot (no spurious 13 GB reload). - Add cleanup_one(key) helper as called for in spec section 4. - Fix pyproject Homepage URL to point at this repo.
25 lines
664 B
Python
25 lines
664 B
Python
"""ComfyUI-MOSS-TTS — Custom nodes for MOSS-TTS 1.7B zero-shot voice cloning."""
|
|
|
|
from .nodes import (
|
|
MOSSTTSLoader,
|
|
MOSSTTSVoiceReference,
|
|
MOSSTTSGenerate,
|
|
MOSSTTSUnload,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"MOSSTTSLoader": MOSSTTSLoader,
|
|
"MOSSTTSVoiceReference": MOSSTTSVoiceReference,
|
|
"MOSSTTSGenerate": MOSSTTSGenerate,
|
|
"MOSSTTSUnload": MOSSTTSUnload,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"MOSSTTSLoader": "MOSS-TTS Loader",
|
|
"MOSSTTSVoiceReference": "MOSS-TTS Voice Reference",
|
|
"MOSSTTSGenerate": "MOSS-TTS Generate",
|
|
"MOSSTTSUnload": "MOSS-TTS Unload",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|