From 2fc38d872d75074c3cc87b27d574ef97800ee97d Mon Sep 17 00:00:00 2001
From: 997146918 <997146918@qq.com>
Date: Tue, 12 Aug 2025 15:38:21 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0all-MiniLM-L6-v2=E6=A8=A1?=
 =?UTF-8?q?=E5=9E=8B=E4=B8=8B=E8=BD=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 AITrain/model_download.py | 3 ++-
 AITrain/requirements.txt  | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/AITrain/model_download.py b/AITrain/model_download.py
index e3b66a2..f6d9fd0 100644
--- 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')
\ No newline at end of file
+model_dir = snapshot_download('Qwen/Qwen3-4B', cache_dir='./', revision='master')
+snapshot_download('sentence-transformers/all-MiniLM-L6-v2', cache_dir='./', revision='master')
\ No newline at end of file
diff --git a/AITrain/requirements.txt b/AITrain/requirements.txt
index 3e6f987..c717db9 100644
--- 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
\ No newline at end of file
+sentence-transformersfaiss-cpu
+fitz
\ No newline at end of file