Fix test_modelopt_export using stale ModelConfig kwargs (#23214)
Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.7
parent
1ebe1c57ed
commit
97baf17557
@@ -321,11 +321,10 @@ class TestModelOptExportIntegration(unittest.TestCase):
|
|||||||
|
|
||||||
model_config = ModelConfig(
|
model_config = ModelConfig(
|
||||||
model_path="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
|
model_path="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
|
||||||
modelopt_quant="fp8",
|
quantization="modelopt_fp8",
|
||||||
modelopt_export_path=self.export_dir,
|
|
||||||
)
|
)
|
||||||
|
|
||||||
load_config = LoadConfig()
|
load_config = LoadConfig(modelopt_export_path=self.export_dir)
|
||||||
device_config = DeviceConfig()
|
device_config = DeviceConfig()
|
||||||
|
|
||||||
# Mock the quantization and export process
|
# Mock the quantization and export process
|
||||||
|
|||||||
Reference in New Issue
Block a user