- Added assertion to ensure weight mapping is not None before checking layer presence in _should_refit_layer method.
- Enhanced error handling in _should_refit_layer to provide clearer feedback when weight mapping is missing.
- Updated assertions in _schedule_device to ensure shape mapping is available for refitting.
- Removed redundant code and organized imports in `trt_loader.py`.
- Consolidated TensorRT search path logic into a dedicated function.
- Updated `TRTDiffuser` and `TRTAnimaDiffuser` classes to streamline model handling.
- Enhanced `ModelBundle` class to improve file handling and error reporting.
- Introduced type hints for better code clarity and maintainability.
- Removed deprecated functions and ensured compatibility with the latest TensorRT and ONNX standards.
zaochuan5854