comfy aimdo 0.2.1

[rattus128]

Changes:

throttle VRAM threshold checks to restore performance in high-layer-rate conditions.
Merge the cuda VRAM consumption estimate with WDDM to mitigate lag in async allocator frees.

Example test conditions:

RTX 5090, PCIE5x16, Ryzen 9600 underclocked to 2GHz. Linux
Ace-step 1.5, 195s

Before:

LM sampling: 100%|██████████| 975/975 [00:23<00:00, 42.36it/s]

After:

LM sampling: 100%|██████████| 975/975 [00:16<00:00, 57.90it/s]

---

RTX5060, Windows WAN 2.1 608x400x81f

No arguments:

loaded partially; 2692.32 MB usable, 2206.05 MB loaded, 11423.82 MB offloaded, 472.63 MB buffer reserved, lowvram patches: 744
100%|██████████████████████████████████████████████████████████████████████████████████████| 4/4 [01:41<00:00, 25.33s/it]
Requested to load WanVAE
loaded completely; 914.76 MB usable, 242.03 MB loaded, full load: True
Prompt executed in 132.54 seconds

With --fast dynamic_vram:

Model WAN21 prepared for dynamic VRAM loading. 13630MB Staged. 1053 patches attached.
100%|██████████████████████████████████████████████████████████████████████████████████████| 4/4 [01:40<00:00, 25.17s/it]
Requested to load WanVAE
0 models unloaded.
Model WanVAE prepared for dynamic VRAM loading. 242MB Staged. 0 patches attached.
Prompt executed in 125.67 seconds

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between b874bd2 and e2e9568.

📒 Files selected for processing (1)

• requirements.txt

---

📝 Walkthrough

Walkthrough

This pull request updates a single dependency constraint in requirements.txt. The comfy-aimdo package version requirement is modified from >=0.2.0 to >=0.2.1, allowing installation of the newer patch version and any compatible future versions. The change consists of one line modified in the dependencies file with no other alterations to the codebase.

🚥 Pre-merge checks | ✅ 3

✅ Passed checks (3 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title 'comfy aimdo 0.2.1' directly corresponds to the single change: updating the comfy-aimdo package to version 0.2.1 in requirements.txt.
Description check	✅ Passed	The description details performance improvements from the comfy-aimdo 0.2.1 update, including VRAM throttling and async allocator optimizations, with benchmark results demonstrating the changes.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}