diff --git a/optimum/onnxruntime/quantization.py b/optimum/onnxruntime/quantization.py
index 0fa6e55f6a3..9b2bc6d4541 100644
--- a/optimum/onnxruntime/quantization.py
+++ b/optimum/onnxruntime/quantization.py
@@ -482,6 +482,7 @@ def quantize(
     def get_calibration_dataset(
         self,
         dataset_name: str,
+        data_files: str = None,
         num_samples: int = 100,
         dataset_config_name: Optional[str] = None,
         dataset_split: Optional[str] = None,
@@ -525,6 +526,7 @@ def get_calibration_dataset(
         calib_dataset = load_dataset(
             dataset_name,
             name=dataset_config_name,
+            data_files=data_files,
             split=dataset_split,
             use_auth_token=use_auth_token,
         )