config is already availableg

huggingface · Apr 11, 2024 · 5c01df0 · 5c01df0
1 parent 4976628
commit 5c01df0
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/src/optimum/nvidia/hub.py b/src/optimum/nvidia/hub.py
@@ -383,17 +383,7 @@ def _from_pretrained(
         except OSError:
             generation_config = None
 
-        transformers_config = AutoConfig.from_pretrained(
-            model_id,
-            revision=revision,
-            cache_dir=cache_dir,
-            force_download=force_download,
-            proxies=proxies,
-            resume_download=resume_download,
-            local_files_only=local_files_only,
-            token=token,
-        )
-
+        transformers_config = AutoConfig.for_model(**config)
         model = cls(
             engines_folders,
             gpus_per_node=model_kwargs.pop("gpus_per_node", 1),