fix

huggingface · Oct 13, 2023 · fe4e8dd · fe4e8dd
1 parent 20268b1
commit fe4e8dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/neural_compressor/test_optimization.py b/tests/neural_compressor/test_optimization.py
@@ -375,7 +375,7 @@ def calibration_fn(p_model):
                 save_directory=tmp_dir,
                 save_onnx_model=False,
             )
-            model = INCModelForCausalLM.from_pretrained(tmp_dir, export=True)
+            model = INCModelForCausalLM.from_pretrained(tmp_dir)
 
         pre_outputs = quantizer._quantized_model.generate(
             **tokens, do_sample=False, num_beams=1, temperature=0.9, min_length=20, max_length=20