rattus 0bfb936ab4 comfy-aimdo 0.2 - Improved pytorch allocator integration (#12557) ... Integrate comfy-aimdo 0.2 which takes a different approach to installing the memory allocator hook. Instead of using the complicated and buggy pytorch MemPool+CudaPluggableAlloctor, cuda is directly hooked making the process much more transparent to both comfy and pytorch. As far as pytorch knows, aimdo doesnt exist anymore, and just operates behind the scenes. Remove all the mempool setup stuff for dynamic_vram and bump the comfy-aimdo version. Remove the allocator object from memory_management and demote its use as an enablment check to a boolean flag. Comfy-aimdo 0.2 also support the pytorch cuda async allocator, so remove the dynamic_vram based force disablement of cuda_malloc and just go back to the old settings of allocators based on command line input.		2026-02-21 10:52:57 -08:00
..
audio_encoders	Reduce RAM usage, fix VRAM OOMs, and fix Windows shared memory spilling with adaptive model loading (#11845)	2026-02-01 01:01:11 -05:00
cldm	Add better error message for common error. (#10846)	2025-11-23 04:55:22 -05:00
comfy_types	feat: add gradient-slider display mode for FLOAT inputs (#12536)	2026-02-20 22:52:32 -08:00
extra_samplers
image_encoders	Add Hunyuan 3D 2.1 Support (#8714)	2025-09-04 20:36:20 -04:00
k_diffusion	ace15: Use dynamic_vram friendly trange (#12409)	2026-02-11 14:53:42 -05:00
ldm	Fix anima LLM adapter forward when manual cast (#12504)	2026-02-17 07:56:44 -08:00
sd1_tokenizer	Silence clip tokenizer warning. (#8934)	2025-07-16 14:42:07 -04:00
t2i_adapter
taesd	Support LTX2 tiny vae (taeltx_2) (#11929)	2026-01-21 23:03:51 -05:00
text_encoders	Small cleanup and try to get qwen 3 work with the text gen. (#12537)	2026-02-19 22:42:28 -05:00
weight_adapter	MPDynamic: force load flux img_in weight (Fixes flux1 canny+depth lora crash) (#12446)	2026-02-15 20:30:09 -05:00
cli_args.py	Reduce RAM usage, fix VRAM OOMs, and fix Windows shared memory spilling with adaptive model loading (#11845)	2026-02-01 01:01:11 -05:00
clip_config_bigg.json
clip_model.py	Support the siglip 2 naflex model as a clip vision model. (#11831)	2026-01-12 17:05:54 -05:00
clip_vision_config_g.json
clip_vision_config_h.json
clip_vision_config_vitl_336_llava.json
clip_vision_config_vitl_336.json
clip_vision_config_vitl.json
clip_vision_siglip2_base_naflex.json	Support the siglip 2 naflex model as a clip vision model. (#11831)	2026-01-12 17:05:54 -05:00
clip_vision_siglip_384.json
clip_vision_siglip_512.json
clip_vision.py	Reduce RAM usage, fix VRAM OOMs, and fix Windows shared memory spilling with adaptive model loading (#11845)	2026-02-01 01:01:11 -05:00
conds.py	Add some warnings and prevent crash when cond devices don't match. (#9169)	2025-08-04 04:20:12 -04:00
context_windows.py	Add handling for vace_context in context windows (#11386)	2025-12-30 14:40:42 -08:00
controlnet.py	Add working Qwen 2512 ControlNet (Fun ControlNet) support (#12359)	2026-02-13 22:23:52 -05:00
diffusers_convert.py
diffusers_load.py
float.py	Optimize nvfp4 lora applying. (#11866)	2026-01-14 00:49:38 -05:00
gligen.py	Remove some useless code. (#8812)	2025-07-06 07:07:39 -04:00
hooks.py	New Year ruff cleanup. (#11595)	2026-01-01 22:06:14 -05:00
latent_formats.py	Basic support for the ace step 1.5 model. (#12237)	2026-02-03 00:06:18 -05:00
lora_convert.py	Use torch RMSNorm for flux models and refactor hunyuan video code. (#12432)	2026-02-13 15:35:13 -05:00
lora.py	MPDynamic: force load flux img_in weight (Fixes flux1 canny+depth lora crash) (#12446)	2026-02-15 20:30:09 -05:00
memory_management.py	comfy-aimdo 0.2 - Improved pytorch allocator integration (#12557)	2026-02-21 10:52:57 -08:00
model_base.py	Fix anima preprocess text embeds not using right inference dtype. (#12501)	2026-02-17 00:29:20 -05:00
model_detection.py	Use torch RMSNorm for flux models and refactor hunyuan video code. (#12432)	2026-02-13 15:35:13 -05:00
model_management.py	comfy-aimdo 0.2 - Improved pytorch allocator integration (#12557)	2026-02-21 10:52:57 -08:00
model_patcher.py	Implement NAG on all the models based on the Flux code. (#12500)	2026-02-16 23:30:34 -05:00
model_sampling.py	Refactor model sampling sigmas code. (#10250)	2025-10-08 17:49:02 -04:00
nested_tensor.py	WIP way to support multi multi dimensional latents. (#10456)	2025-10-23 21:21:14 -04:00
ops.py	ops: limit return of requants (#12506)	2026-02-17 15:32:27 -05:00
options.py
patcher_extension.py	Fix order of inputs nested merge_nested_dicts (#10362)	2025-10-15 16:47:26 -07:00
pinned_memory.py	fix pinning with model defined dtype (#12208)	2026-02-01 08:42:32 -08:00
pixel_space_convert.py	Changes to the previous radiance commit. (#9851)	2025-09-13 18:03:34 -04:00
quant_ops.py	Optimize nvfp4 lora applying. (#11866)	2026-01-14 00:49:38 -05:00
rmsnorm.py	Remove code to support RMSNorm on old pytorch. (#12499)	2026-02-16 20:09:24 -05:00
sample.py	Make regular empty latent node work properly on flux 2 variants. (#12050)	2026-01-23 19:50:48 -05:00
sampler_helpers.py	[Trainer] training with proper offloading (#12189)	2026-02-10 21:45:19 -05:00
samplers.py	Reduce RAM usage, fix VRAM OOMs, and fix Windows shared memory spilling with adaptive model loading (#11845)	2026-02-01 01:01:11 -05:00
sd1_clip_config.json
sd1_clip.py	Force min length 1 when tokenizing for text generation. (#12538)	2026-02-19 22:57:44 -05:00
sd.py	Small cleanup and try to get qwen 3 work with the text gen. (#12537)	2026-02-19 22:42:28 -05:00
sdxl_clip.py
supported_models_base.py	Fix some custom nodes. (#11134)	2025-12-05 18:25:31 -05:00
supported_models.py	Use torch RMSNorm for flux models and refactor hunyuan video code. (#12432)	2026-02-13 15:35:13 -05:00
utils.py	comfy-aimdo 0.2 - Improved pytorch allocator integration (#12557)	2026-02-21 10:52:57 -08:00
windows.py	Reduce RAM usage, fix VRAM OOMs, and fix Windows shared memory spilling with adaptive model loading (#11845)	2026-02-01 01:01:11 -05:00