bemoregtandClaude Sonnet 4.6
9a0c3dc1a0
- LoadFFTNetModel / FFTNetGenerate ComfyUI nodes - FFTNet architecture (SpectralFilter, LocalWindowMixing, FFTNetMixer) - upgrade/: replace LLM self-attention with FFTNet + MPS-safe fine-tuning script - Auto-detects self_attn return count (1/2/3) for transformers version compat - MPS FFT CPU offload workaround for Apple Silicon Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
14 lines
342 B
Python
14 lines
342 B
Python
from .nodes import LoadFFTNetModel, FFTNetGenerate
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"LoadFFTNetModel": LoadFFTNetModel,
|
|
"FFTNetGenerate": FFTNetGenerate,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"LoadFFTNetModel": "Load FFTNet Model",
|
|
"FFTNetGenerate": "FFTNet Generate",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|