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
ajrasane
41f2bf4941
Add option to real quantize the model ( #473 )
...
Signed-off-by: ajrasane <131806219+ajrasane@users.noreply.github.com >
2025-10-28 13:29:19 -07:00
vishalpandya1990
99c76fff26
Add SD3.5-medium quantization support in ModelOpt Diffusers example ( #444 )
...
Signed-off-by: vipandya <vipandya@nvidia.com >
2025-10-22 23:28:15 -07:00
ajrasane
2fd67cc711
Add option to benchmark pipeline in diffusion_trt.py ( #457 )
...
Signed-off-by: ajrasane <131806219+ajrasane@users.noreply.github.com >
2025-10-22 15:32:45 -07:00
Keval Morabia
e6e0d2cc8f
Fix failing CICD nightly tests ( #445 )
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-10-20 19:16:45 +00:00
jingyu-ml and Keval Morabia
46a9e49ab3
Fixed cache diffusion ci/cd ( #426 )
...
Signed-off-by: Jingyu Xin <jingyux@nvidia.com >
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
Co-authored-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-10-13 16:31:55 +05:30
Keval Morabia
c0590b0255
Deprecate ModelOpt custom docker and directly use TRT-LLM / PyTorch / TRT docker ( #346 )
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-09-23 01:43:25 +05:30
jingyu-ml
358b0c61b4
Fixed the CICD for diffusers ( #312 )
...
Signed-off-by: jingyu <jingyu@omniml.ai >
Signed-off-by: Jingyu Xin <jingyux@nvidia.com >
2025-09-10 17:03:03 -05:00
Keval Morabia
1ef1d72a1b
Code quality improvements - typos, formatting, etc.
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-09-02 19:58:33 +05:30
Keval Morabia
4d1eb0caf5
Major improvement of READMEs and documentation
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-08-30 10:54:28 +05:30
Keval Morabia
d09c82779c
Push latest changes and bug fixes ( #274 )
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-08-29 17:29:52 +05:30
Keval Morabia
dcdc2df084
Push latest changes
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-08-26 01:12:26 +05:30
Keval Morabia
4c611e47a6
Update files on GitHub
...
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com >
2025-07-31 22:48:09 +05:30
Keval Morabia
cafa7f60ed
Update for 0.33.0 release
2025-07-14 23:08:26 +05:30
Keval Morabia
7af33d29ce
Update for 0.31.0 release
2025-06-05 13:24:07 -07:00
Keval Morabia
7a047435ae
Update for 0.29.0 release
2025-05-08 23:43:51 +05:30
Keval Morabia
9cb36cb43c
Update files for 0.27.1 release
2025-04-15 23:36:35 +05:30
Keval Morabia
92f430f6ab
Add files for 0.27.0 release
2025-04-03 10:31:41 +05:30
Keval Morabia
2017cd9063
Update for 0.25.0 release
2025-03-03 22:54:22 +05:30
Keval Morabia
73d6af785f
Update 0.23.0 - OSS release
2025-01-29 01:45:47 +05:30
ajrasane