WillHeld
/

DiVA-llama-3-v0-8b

Feature Extraction

Model card Files Files and versions Community

Helw150 commited on Jun 20, 2024

Commit

fcf1cf7

•

1 Parent(s): cd5984d

Need a Filename

Files changed (1) hide show

modeling_diva.py +1 -0

modeling_diva.py CHANGED Viewed

@@ -126,6 +126,7 @@ class DiVAModel(PreTrainedModel):
             hf_hub_download(
                 repo_id=pretrained_model_name_or_path,
                 token=kwargs.get("token", None),
                 local_dir=os.path.dirname(__file__),
             )

             hf_hub_download(
                 repo_id=pretrained_model_name_or_path,
+                filename="model-00001-of-00004.safetensors",
                 token=kwargs.get("token", None),
                 local_dir=os.path.dirname(__file__),
             )