WillHeld
/

DiVA-llama-3-v0-8b

Feature Extraction

Model card Files Files and versions Community

WillHeld commited on 20 days ago

Commit

ef73e7d

•

1 Parent(s): 79fec82

Update modeling_diva.py

Files changed (1) hide show

modeling_diva.py +2 -2

modeling_diva.py CHANGED Viewed

@@ -139,13 +139,13 @@ class DiVAModel(PreTrainedModel):
                 repo_id=pretrained_model_name_or_path,
                 filename="model.safetensors",
                 token=kwargs.get("token", None),
-                local_dir=os.path.dirname(__file__),
             )
             hf_hub_download(
                 repo_id=pretrained_model_name_or_path,
                 filename="config.json",
                 token=kwargs.get("token", None),
-                local_dir=os.path.dirname(__file__),
             )
             via_path = os.path.dirname(__file__) + "/model.safetensors"
             config_path = os.path.dirname(__file__) + "/config.json"

                 repo_id=pretrained_model_name_or_path,
                 filename="model.safetensors",
                 token=kwargs.get("token", None),
+                cache_dir=os.path.dirname(__file__),
             )
             hf_hub_download(
                 repo_id=pretrained_model_name_or_path,
                 filename="config.json",
                 token=kwargs.get("token", None),
+                cache_dir=os.path.dirname(__file__),
             )
             via_path = os.path.dirname(__file__) + "/model.safetensors"
             config_path = os.path.dirname(__file__) + "/config.json"