diff --git a/.idea/.gitignore b/.idea/.gitignore
new file mode 100644
index 0000000..26d3352
--- /dev/null
+++ b/.idea/.gitignore
@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml
diff --git a/.idea/ComfyUI-MuseTalk-KJ.iml b/.idea/ComfyUI-MuseTalk-KJ.iml
new file mode 100644
index 0000000..039314d
--- /dev/null
+++ b/.idea/ComfyUI-MuseTalk-KJ.iml
@@ -0,0 +1,12 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+  <component name="PyDocumentationSettings">
+    <option name="format" value="NUMPY" />
+    <option name="myDocStringFormat" value="NumPy" />
+  </component>
+</module>
\ No newline at end of file
diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml
new file mode 100644
index 0000000..105ce2d
--- /dev/null
+++ b/.idea/inspectionProfiles/profiles_settings.xml
@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>
\ No newline at end of file
diff --git a/.idea/modules.xml b/.idea/modules.xml
new file mode 100644
index 0000000..0ebb300
--- /dev/null
+++ b/.idea/modules.xml
@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/ComfyUI-MuseTalk-KJ.iml" filepath="$PROJECT_DIR$/.idea/ComfyUI-MuseTalk-KJ.iml" />
+    </modules>
+  </component>
+</project>
\ No newline at end of file
diff --git a/.idea/vcs.xml b/.idea/vcs.xml
new file mode 100644
index 0000000..35eb1dd
--- /dev/null
+++ b/.idea/vcs.xml
@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>
\ No newline at end of file
diff --git a/nodes.py b/nodes.py
index cb1878c..e9f180d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -43,6 +43,9 @@ def __init__(self):
         self.latent_format = comfy.latent_formats.SD15
         self.manual_cast_dtype = None
         self.sampling_settings = {}
+        self.memory_usage_factor = 1.0
+        self.custom_operations = None
+        self.optimizations = {}
 
 class UNETLoader_MuseTalk:
     @classmethod
@@ -308,4 +311,4 @@ def get_sliced_feature(self,feature_array, vid_idx, audio_feat_length= [2,2],fps
     "vhs_audio_to_audio_tensor": "VHS Audio To Audio Tensor",
     "muse_talk_sampler": "MuseTalk Sampler",
     "UNETLoader_MuseTalk": "UNETLoader_MuseTalk"
-}
+}
\ No newline at end of file