[tests] migrate model-level ModelOpt quantization tests to model tests - #14256
Open
sayakpaul wants to merge 3 commits into
Open
[tests] migrate model-level ModelOpt quantization tests to model tests#14256sayakpaul wants to merge 3 commits into
sayakpaul wants to merge 3 commits into
Conversation
… mixins Move the model-level tests out of tests/quantization/modelopt/test_modelopt.py (the ModelOptBaseTesterMixin subclasses) into the standardized model-test mixin structure: - Extend ModelOptTesterMixin with keep_modules_in_fp32, training and prequantized-serialization coverage, and set disable_conv_quantization in the shared MODELOPT_CONFIGS so conv-containing models (SD3 patch embed) don't break. - Wire concrete SD3 classes: TestSD3TransformerModelOpt and TestSD3TransformerModelOptCompile. - Trim test_modelopt.py to the remaining pipeline-level test_model_cpu_offload (kept per quant type, incl. NF4/NVFP4). Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
github-actions
Bot
added
size/L
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
As discussed in #14113 (comment)
Also filed a related issue: NVIDIA/Model-Optimizer#2001