Model-Optimizer

mirror of https://github.com/NVIDIA/Model-Optimizer.git synced 2026-11-09 09:14:17 +08:00

Files

T

ajrasane 229053323a [NVBUG: 5619158] Enforce high precision model dtype for diffusion trt (#526 )

## What does this PR do?

**Type of change:** 
Minor code change

**Overview:** 
- Select the high precision dtype directly based on model type - FP16
for Stable Diffusion models, BF16 for Flux


## Testing
```python
python diffusion_trt.py --model flux-dev --benchmark
```

## Before your PR is "*Ready for review*"
<!-- If you haven't finished some of the above items you can still open
`Draft` PR. -->

- **Make sure you read and follow [Contributor
guidelines](https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/main/CONTRIBUTING.md)**
and your commits are signed.
- **Is this change backward compatible?**: No (No option to specify
dtype while loading pipeline)
- **Did you write any new necessary tests?**: No
- **Did you add or update any necessary documentation?**: Yes
- **Did you update
[Changelog](https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/main/CHANGELOG.rst)?**:
Yes

---------

Signed-off-by: ajrasane <131806219+ajrasane@users.noreply.github.com>

2025-11-06 23:44:37 -08:00

assets

Update 0.23.0 - OSS release

2025-01-29 01:45:47 +05:30

calib

Push latest changes

2025-08-26 01:12:26 +05:30

onnx_utils

Add SD3.5-medium quantization support in ModelOpt Diffusers example (#444 )

2025-10-22 23:28:15 -07:00

.gitignore

Update 0.23.0 - OSS release