diff --git a/nodes/gpu/comfy-env.toml b/nodes/gpu/comfy-env.toml
index 7bdf454f..7e5eae08 100644
--- a/nodes/gpu/comfy-env.toml
+++ b/nodes/gpu/comfy-env.toml
@@ -7,6 +7,8 @@ trimesh = { version = ">=4.0.0", extras = ["easy"] }
 tqdm = "*"
 psutil = "*"
 safetensors = "*"
+comfy-aimdo = "*"
+einops = "*"
 
 [target.win-64.pypi-dependencies]
 msvc-runtime = "*"