feat: expose feature bank to comfyui (#12)

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>

src/__init__.py

@@ -0,0 +1 @@
+# Dummy file ensuring this sub-package will be recognized.

src/feature_bank/__init__.py

@@ -0,0 +1 @@
+# Dummy file ensuring this sub-package will be recognized.

src/feature_bank/feature_bank.py

@@ -191,7 +191,7 @@ class FeatureBankAttentionProcessor:
                 batch_size, channel, height * width
             ).transpose(1, 2)
 
-        batch_size, _ , _ = (
+        batch_size, _, _ = (
             hidden_states.shape
             if encoder_hidden_states is None
             else encoder_hidden_states.shape
@@ -459,7 +459,7 @@ class FeatureBankAttentionProcessorNode:
 
     RETURN_TYPES = ("MODEL",)
     FUNCTION = "update_node"
-    CATEGORY = "feature-bank"
+    CATEGORY = "StreamPack/model_patches/unet"
 
     def update_node(
         self,