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

2025-08-12 15:38:21 +08:00 · 2025-08-12 15:38:21 +08:00 · 2fc38d872d
commit 2fc38d872d
parent ca4dcb5386
2 changed files with 4 additions and 2 deletions
--- a/AITrain/model_download.py
+++ b/AITrain/model_download.py
@ -1,2 +1,3 @@
 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')
--- a/AITrain/requirements.txt
+++ b/AITrain/requirements.txt
@ -16,4 +16,5 @@ datasets==3.5.1
 peft==0.15.2
 autoawq
 PyPDF2
-sentence-transformersfaiss-cpu
+sentence-transformersfaiss-cpu
+fitz