changing variable name

deepanker13 · Dec 8, 2023 · eb7c9d0 · eb7c9d0
1 parent afc409f
commit eb7c9d0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sdk/python/kubeflow/storage_init_container/hugging_face.py b/sdk/python/kubeflow/storage_init_container/hugging_face.py
@@ -21,7 +21,7 @@ class HuggingFaceModelParams:
     access_token: str
     model_uri: str
     transformer_type: Literal[*TRANSFORMER_TYPES]
-    mount_path: str = field(default="/workspace/models")
+    download_dir: str = field(default="/workspace/models")
 
     def __post_init__(self):
         # Custom checks or validations can be added here
@@ -48,9 +48,9 @@ def download_model_and_tokenizer(self):
         transformer_type_class.from_pretrained(
             self.model,
             token=self.config.access_token,
-            cache_dir=self.config.mount_path,
+            cache_dir=self.config.download_dir,
             trust_remote_code=True,
         )
         transformers.AutoTokenizer.from_pretrained(
-            self.model, cache_dir=self.config.mount_path
+            self.model, cache_dir=self.config.download_dir
         )