diff --git a/mypy.ini b/mypy.ini
index 3894de4..64e882d 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -15,6 +15,10 @@ warn_no_return = True
 warn_unreachable = True
 strict_equality = True
 
+# Main module
+[mypy-__main__.*]
+ignore_missing_imports = True
+
 # ComfyUI folder_paths module
 [mypy-folder_paths]
 ignore_missing_imports = True