fix:修复docker镜像构建问题
This commit is contained in:
@@ -13,6 +13,8 @@ logger = get_logger(__name__)
|
||||
# 配置离线模式,避免联网检查
|
||||
os.environ['TRANSFORMERS_OFFLINE'] = '1'
|
||||
os.environ['HF_DATASETS_OFFLINE'] = '1'
|
||||
os.environ['HF_HUB_OFFLINE'] = '1' # 禁用HuggingFace Hub联网
|
||||
os.environ['SENTENCE_TRANSFORMERS_HOME'] = 'data/models' # 设置模型缓存目录
|
||||
|
||||
|
||||
class MemoryService:
|
||||
|
||||
@@ -19,6 +19,15 @@ anthropic==0.72.0
|
||||
httpx==0.28.1
|
||||
python-dotenv==1.0.0
|
||||
|
||||
# NumPy版本锁定(兼容性要求)
|
||||
numpy==1.26.4
|
||||
|
||||
# 向量数据库和Embedding (长期记忆系统)
|
||||
chromadb==1.3.2
|
||||
sentence-transformers>=2.3.1
|
||||
|
||||
|
||||
# Transformers(锁定兼容版本)
|
||||
transformers==4.35.2
|
||||
|
||||
# Sentence Transformers(基于PyTorch的文本embedding库)
|
||||
sentence-transformers==2.3.1
|
||||
Reference in New Issue
Block a user