trending/Model-Optimizer
1
0
Fork 0
mirror of https://github.com/NVIDIA/Model-Optimizer.git synced 2026-11-09 09:14:17 +08:00
Code Issues Packages Projects Releases Wiki Activity
240 Commits 468 Branches 62 Tags
fc92e98fe9635b5380140ddbd2a7efa80e46ab81
Commit Graph
26 Commits
Author SHA1 Message Date
Gal Hubara-Agam ae915eaea9 [OMNIML-2493] AutoCast: Configure target opset from CLI (#519) 
## What does this PR do?

new feature

**Overview:** 

AutoCast: Allow configuring target opset from CLI and main API.

## Usage
<!-- You can potentially add a usage example below. -->
python -m modelopt.onnx.autocast --onnx_path model.onnx --opset 22 

```python
# Add a code snippet demonstrating how to use this
```

## Testing
<!-- Mention how have you tested your change if applicable. -->
pytest tests/unit/onnx/autocast/test_autocast.py -k test_opset

## 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?**: Yes
- **Did you write any new necessary tests?**: Yes
- **Did you add or update any necessary documentation?**: Yes
- **Did you update
[Changelog](https://github.com/NVIDIA/TensorRT-Model-Optimizer/blob/main/CHANGELOG.rst)?**:
No - minor API addition <!--- Only for new features, API changes,
critical bug fixes or bw breaking changes. -->

## Additional Information
<!-- E.g. related issue. -->

Signed-off-by: Gal Hubara Agam <96368689+galagam@users.noreply.github.com>
 2025-11-06 19:13:09 +02:00
Gal Hubara-Agam 7b6a15a004 [5545101]: AutoCast: Add options to force include node/op in F16 (#386) 
Signed-off-by: Gal Hubara Agam <96368689+galagam@users.noreply.github.com>
 2025-10-28 19:59:15 +00:00
Keval Morabia 90b1e68fcf Update HF nvidia collection links in docs (#475) 
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-10-28 22:23:01 +05:30
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
Zhiyu bc546943b4 Add a deprecation notice for trtllm export in doc (#410) 
Signed-off-by: Zhiyu Cheng <zhiyuc@nvidia.com>
 2025-10-09 05:22:29 +00:00
Keval Morabia 3a76d28f5b Fix multi-gpu mamba minitron param sorting test + minor cleanup (#412) 
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-10-08 22:09:53 +05:30
Mohammed Yasin 70abfb416f Fix FLOPs calculation (#388) 
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Signed-off-by: Y-T-G <32206511+Y-T-G@users.noreply.github.com>
 2025-09-30 21:40:51 +05:30
Yuening Li d649fb8375 Support INT8 Weight-Only Quantization (#263) 
Signed-off-by: Yuening Li <62227368+Yuening-wa@users.noreply.github.com>
 2025-09-26 17:46:30 +00:00
Chenjie LuoandKeval Morabia ad091e884e Improve realquant gemm impl (#368) 
Signed-off-by: Chenjie Luo <chenjiel@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-09-26 09:37:51 +05:30
Keval Morabia b4d6cedb4a Support checkpointing Minitron pruning scores / prune without sorting (#361) 
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-09-26 01:46:22 +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
Keval Morabia 4c36abe536 Add llm_ptq PR test, Cleanup dependency installation in modelopt docker (#338) 
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-09-19 10:48:44 +05:30
Asha Anoosheh b895dc5c7c Disable KD mode from saving problematic state (#320) 
Signed-off-by: Asha Anoosheh <aanoosheh@nvidia.com>
 2025-09-19 02:31:20 +05:30
Chenjie LuoChenjie LuoKeval Morabia
 		146d1d9d40 Deprecate TRTLLM-build in examples (#297) 
Signed-off-by: Chenjie Luo <chenjiel@nvidia.com>
Signed-off-by: Chenjie Luo <108829653+cjluo-nv@users.noreply.github.com>
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
Co-authored-by: Chenjie Luo <chenjie@omniml.ai>
Co-authored-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-09-18 03:22:37 +05:30
Keval Morabia 103b1bb137 Push latest changes 
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-09-04 11:02:58 +05:30
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 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 37085354c9 Update files on Github (#258) 
Signed-off-by: Keval Morabia <28916987+kevalmorabia97@users.noreply.github.com>
 2025-08-17 02:44:16 +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 e048fb239d Add changes for 0.27 Windows release 2025-04-29 22:11:52 +05:30
Keval Morabia 92f430f6ab Add files for 0.27.0 release 2025-04-03 10:31:41 +05:30
Keval Morabia 7eecd11b0f Add source code for 0.25.0 docs 2025-03-03 23:02:22 +05:30