zaochuan5854 6d818154fb fix(trt_diffusers): temporary fallback to CPU to avoid GPU error #21 ...

- Force CPU fallback in AnimaONNXWrapper and trt_exporter as a temporary workaround for unexplained GPU error #21.
- Update target and source attention masks to support 4D shapes and dimensions.
- Enforce contiguous memory layout for all inference inputs and output buffers.
- Adjust trt_exporter to run ONNX tracing on CPU with float32 for bf16 parameter conversion.

Closes #21

2026-05-29 18:36:54 +09:00

..

anima_diffuser.py

fix(trt_diffusers): temporary fallback to CPU to avoid GPU error #21

2026-05-29 18:36:54 +09:00

base_diffuser.py

refactor: streamline TRTAnimaDiffuser and TRTDiffuser constructors

2026-04-23 18:39:36 +09:00