Rick StaaandRUFFY-369
eedf5d1d1b
This commit establishes the required file structure and logic to integrate the Feature Bank node into ComfyUI, enabling its use. --------- Co-authored-by: RUFFY-369 <prakarshkaushik369@gmail.com>
21 lines
594 B
Python
21 lines
594 B
Python
"""A ComfyUI native node implementation of the Feature Bank of the StreamV2V paper.
|
|
|
|
|
|
References:
|
|
StreamV2V: A Deep Generative Model for High-fidelity View Synthesis - Liang et al.,
|
|
2025, URL: https://arxiv.org/abs/2405.15757
|
|
"""
|
|
|
|
from ..src.feature_bank.feature_bank import (
|
|
FeatureBankAttentionProcessorNode,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FeatureBankAttentionProcessor": FeatureBankAttentionProcessorNode
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"FeatureBankAttentionProcessor": "Feature Bank Attention Processor"
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|