From 1d6797c49a62e3512ed6fa36129199d6d680fb36 Mon Sep 17 00:00:00 2001
From: "Wang, Chang" <chang1.wang@intel.com>
Date: Thu, 29 Aug 2024 10:53:21 +0800
Subject: [PATCH] Update optimum/intel/neural_compressor/modeling_base.py

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
---
 optimum/intel/neural_compressor/modeling_base.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/optimum/intel/neural_compressor/modeling_base.py b/optimum/intel/neural_compressor/modeling_base.py
index 4d718e81e7..6a0d0499da 100644
--- a/optimum/intel/neural_compressor/modeling_base.py
+++ b/optimum/intel/neural_compressor/modeling_base.py
@@ -151,7 +151,13 @@ def _from_pretrained(
                     _BaseQBitsAutoModelClass.ORIG_MODEL = cls.auto_model_class
                     model = _BaseQBitsAutoModelClass.load_low_bit(
                         model_id,
-                        *model_args,
+                        subfolder=subfolder,
+                        revision=revision,
+                        cache_dir=cache_dir,
+                        token=token,
+                        local_files_only=local_files_only,
+                        force_download=force_download,
+                        trust_remote_code=trust_remote_code,
                         config=config,
                         **kwargs,
                     )