admin/Project02
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

添加all-MiniLM-L6-v2模型下载

Browse Source
This commit is contained in:
997146918 2025-08-12 15:38:21 +08:00
parent ca4dcb5386
commit 2fc38d872d
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
AITrain/model_download.py
View File

@ -1,2 +1,3 @@
from modelscope import snapshot_download from modelscope import snapshot_download
model_dir = snapshot_download('deepseek-ai/DeepSeek-R1-Distill-Qwen-7B', cache_dir='./', revision='master') model_dir = snapshot_download('Qwen/Qwen3-4B', cache_dir='./', revision='master')
snapshot_download('sentence-transformers/all-MiniLM-L6-v2', cache_dir='./', revision='master')

3
AITrain/requirements.txt
View File

@ -16,4 +16,5 @@ datasets==3.5.1
peft==0.15.2 peft==0.15.2
autoawq autoawq
PyPDF2 PyPDF2
sentence-transformersfaiss-cpu sentence-transformersfaiss-cpu
fitz