rattus 535c16ce6e Widen OOM_EXCEPTION to AcceleratorError form (#12835) ...

Pytorch only filters for OOMs in its own allocators however there are
paths that can OOM on allocators made outside the pytorch allocators.
These manifest as an AllocatorError as pytorch does not have universal
error translation to its OOM type on exception. Handle it. A log I have
for this also shows a double report of the error async, so call the
async discarder to cleanup and make these OOMs look like OOMs.

2026-03-10 00:41:02 -04:00

..

diffusionmodules

Widen OOM_EXCEPTION to AcceleratorError form (#12835)

2026-03-10 00:41:02 -04:00

distributions

Small optimizations.

2024-12-18 18:23:28 -05:00

encoders

Make unclip more deterministic.

2024-01-14 17:28:31 -05:00

attention.py

Widen OOM_EXCEPTION to AcceleratorError form (#12835)

2026-03-10 00:41:02 -04:00

ema.py

New Year ruff cleanup. (#11595)

2026-01-01 22:06:14 -05:00

sdpose.py

feat: Support SDPose-OOD (#12661)

2026-02-26 19:59:05 -05:00

sub_quadratic_attention.py

Widen OOM_EXCEPTION to AcceleratorError form (#12835)

2026-03-10 00:41:02 -04:00

temporal_ae.py

Basic Hunyuan Video model support.

2024-12-16 19:35:40 -05:00