github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-08-08 02:53:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update setup script to point to the new settings

Browse Source
This commit is contained in:
imartinez 2024-03-01 09:15:43 +01:00
parent c3fe36e070
commit 8fad1966f0
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
scripts/setup
View File

@ -19,19 +19,19 @@ os.makedirs(models_path, exist_ok=True)
# Download Embedding model # Download Embedding model
embedding_path = models_path / "embedding" embedding_path = models_path / "embedding"
print(f"Downloading embedding {settings().local.embedding_hf_model_name}") print(f"Downloading embedding {settings().huggingface.embedding_hf_model_name}")
snapshot_download( snapshot_download(
repo_id=settings().local.embedding_hf_model_name, repo_id=settings().huggingface.embedding_hf_model_name,
cache_dir=models_cache_path, cache_dir=models_cache_path,
local_dir=embedding_path, local_dir=embedding_path,
) )
print("Embedding model downloaded!") print("Embedding model downloaded!")
# Download LLM and create a symlink to the model file # Download LLM and create a symlink to the model file
print(f"Downloading LLM {settings().local.llm_hf_model_file}") print(f"Downloading LLM {settings().llamacpp.llm_hf_model_file}")
hf_hub_download( hf_hub_download(
repo_id=settings().local.llm_hf_repo_id, repo_id=settings().llamacpp.llm_hf_repo_id,
filename=settings().local.llm_hf_model_file, filename=settings().llamacpp.llm_hf_model_file,
cache_dir=models_cache_path, cache_dir=models_cache_path,
local_dir=models_path, local_dir=models_path,
resume_download=resume_download, resume_download=resume_download,