Rattus 469d7a62de execution: add aimdo primary pytorch cache integration ... We need to general pytorch cache defragmentation on an appropriate level for aimdo. Do in here on the per node basis, which has a reasonable chance of purging stale shapes out of the pytorch caching allocator and saving VRAM without costing too much garbage collector thrash. This looks like a lot of GC but because aimdo never fails from pytorch and saves the pytorch allocator from ever need to defrag out of demand, but it needs a oil change every now and then so we gotta do it. Doing it here also means the pytorch temps are cleared from task manager VRAM usage so user anxiety can go down a little when they see their vram drop back at the end of workflows inline with inference usage (rather than assuming full VRAM leaks).		2026-01-27 18:56:10 +10:00
..
audio_encoders	models: Use CoreModelPatcher	2026-01-27 18:56:10 +10:00
cldm	Add better error message for common error. (#10846)	2025-11-23 04:55:22 -05:00
comfy_types	LoRA Trainer: LoRA training node in weight adapter scheme (#8446)	2025-06-13 19:25:59 -04:00
extra_samplers
image_encoders	Add Hunyuan 3D 2.1 Support (#8714)	2025-09-04 20:36:20 -04:00
k_diffusion	Fix noise with ancestral samplers when inferencing on cpu. (#11528)	2025-12-26 22:03:01 -05:00
ldm	models: Use CoreModelPatcher	2026-01-27 18:56:10 +10: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	Fix mistral 3 tokenizer code failing on latest transformers version and other breakage. (#12095)	2026-01-26 11:39:00 -05:00
weight_adapter	[Weight-adapter/Trainer] Bypass forward mode in Weight adapter system (#11958)	2026-01-24 22:56:22 -05:00
checkpoint_pickle.py
cli_args.py	Add most basic Asset support for models (#11315)	2026-01-08 22:21:51 -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	Support llava clip vision model.	2025-03-06 00:24:43 -05:00
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	Support 512 siglip model.	2025-04-05 07:01:01 -04:00
clip_vision.py	models: Use CoreModelPatcher	2026-01-27 18:56:10 +10: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	models: Use CoreModelPatcher	2026-01-27 18:56:10 +10: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	Make empty latent node work with other models. (#12062)	2026-01-24 19:23:20 -05:00
lora_convert.py	Implement the USO subject identity lora. (#9674)	2025-09-01 18:54:02 -04:00
lora.py	Support ModelScope-Trainer/DiffSynth LoRA format for Flux.2 Klein models (#12042)	2026-01-23 15:27:49 -05:00
memory_management.py	execution: add aimdo primary pytorch cache integration	2026-01-27 18:56:10 +10:00
model_base.py	models: Use CoreModelPatcher	2026-01-27 18:56:10 +10:00
model_detection.py	Only enable fp16 on z image models that actually support it. (#12065)	2026-01-24 22:32:28 -05:00
model_management.py	ops/mp: implement aimdo	2026-01-27 18:56:10 +10:00
model_patcher.py	ops/mp: implement aimdo	2026-01-27 18:56:10 +10: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/mp: implement aimdo	2026-01-27 18:56:10 +10: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	pinned_memory: add python	2026-01-27 18:56:10 +10: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	Add warning when using old pytorch. (#9347)	2025-08-15 00:22:26 -04: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	skip_load_model -> force_full_load (#11390)	2025-12-17 23:29:32 -05:00
samplers.py	mp: wrap get_free_memory	2026-01-27 18:56:10 +10:00
sd1_clip_config.json
sd1_clip.py	Fix mistral 3 tokenizer code failing on latest transformers version and other breakage. (#12095)	2026-01-26 11:39:00 -05:00
sd.py	models: Use CoreModelPatcher	2026-01-27 18:56:10 +10:00
sdxl_clip.py	Add a T5TokenizerOptions node to set options for the T5 tokenizer. (#7803)	2025-04-25 19:36:00 -04:00
supported_models_base.py	Fix some custom nodes. (#11134)	2025-12-05 18:25:31 -05:00
supported_models.py	Only enable fp16 on z image models that actually support it. (#12065)	2026-01-24 22:32:28 -05:00
utils.py	move string_to_seed to utils.py	2026-01-27 18:56:10 +10:00