Update app.py
Browse files
app.py
CHANGED
@@ -55,8 +55,10 @@ repo_name = "IlyaGusev/saiga_13b_lora_llamacpp"
|
|
55 |
model_name = "ggml-model-q4_1.bin"
|
56 |
embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
|
57 |
|
58 |
-
|
59 |
-
snapshot_download(repo_id=
|
|
|
|
|
60 |
|
61 |
model = Llama(
|
62 |
model_path=f"/data/saiga_13b_lora_llamacpp/{model_name}",
|
|
|
55 |
model_name = "ggml-model-q4_1.bin"
|
56 |
embedder_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
|
57 |
|
58 |
+
print("Downloading all files...")
|
59 |
+
snapshot_download(repo_id=repo_name, local_dir="/data/saiga_13b_lora_llamacpp", allow_patterns=model_name, cache_dir="/data")
|
60 |
+
snapshot_download(repo_id=embedder_name, local_dir="/data/paraphrase-multilingual-mpnet-base-v2", cache_dir="/data")
|
61 |
+
print("Files downloaded!")
|
62 |
|
63 |
model = Llama(
|
64 |
model_path=f"/data/saiga_13b_lora_llamacpp/{model_name}",
|