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_path="TinyLlama/TinyLlama-1.1B-Chat-v1.0",
|
||||
modelopt_quant="fp8",
|
||||
modelopt_export_path=self.export_dir,
|
||||
quantization="modelopt_fp8",
|
||||
)
|
||||
|
||||
load_config = LoadConfig()
|
||||
load_config = LoadConfig(modelopt_export_path=self.export_dir)
|
||||
device_config = DeviceConfig()
|
||||
|
||||
# Mock the quantization and export process
|
||||
|
||||
Reference in New Issue
Block a user