Rick Staa
87e609d13a
This commit sets up the necessary files to integrate the Feature Bank node into ComfyUI, making it available for use.
22 lines
595 B
Python
22 lines
595 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"]
|