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

chained_optimizations

ONNX save fix (#496 )

2025-11-05 09:18:56 +05:30

cnn_qat

Add llm_ptq PR test, Cleanup dependency installation in modelopt docker (#338 )

2025-09-19 10:48:44 +05:30

deepseek

Push latest changes

2025-09-04 11:02:58 +05:30

diffusers

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