Spaces:

seanpedrickcase
/

topic_modelling

Running

seanpedrickcase commited on Aug 13

Commit

1c0bfd4

•

1 Parent(s): 51ba1cb

Updated download_model.py to download pytorch .bin file

Files changed (1) hide show

download_model.py CHANGED Viewed

@@ -4,8 +4,7 @@ from huggingface_hub import hf_hub_download
 repo_id = "sentence-transformers/all-MiniLM-L6-v2" #"BAAI/bge-small-en-v1.5"
 files_to_download = [
     "config.json",
-    "config_sentence_transformers.json",
-    "model.safetensors",
     "tokenizer_config.json",
     "vocab.txt"
 ]

 repo_id = "sentence-transformers/all-MiniLM-L6-v2" #"BAAI/bge-small-en-v1.5"
 files_to_download = [
     "config.json",
+    "pytorch_model.bin",
     "tokenizer_config.json",
     "vocab.txt"
 ]