Spaces:
Sleeping
Sleeping
fix: 修复CLI认证环境变量名称
Browse files- 统一使用 Assitant_tocken 作为环境变量名
- 在Dockerfile构建时进行HF认证
- 确保所有组件使用相同的变量名
- Dockerfile +6 -0
- modules/ai_model.py +2 -2
Dockerfile
CHANGED
|
@@ -26,6 +26,12 @@ RUN pip install --no-cache-dir -r requirements.txt
|
|
| 26 |
# 复制项目文件
|
| 27 |
COPY . .
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
# 暴露端口
|
| 30 |
EXPOSE 7860
|
| 31 |
|
|
|
|
| 26 |
# 复制项目文件
|
| 27 |
COPY . .
|
| 28 |
|
| 29 |
+
#hf_cli authenticate
|
| 30 |
+
ARG Assitant_tocken
|
| 31 |
+
RUN if [ -n "$Assitant_tocken" ]; then \
|
| 32 |
+
python -c "from huggingface_hub import login; login('$Assitant_tocken', add_to_git_credential=False)"; \
|
| 33 |
+
fi
|
| 34 |
+
|
| 35 |
# 暴露端口
|
| 36 |
EXPOSE 7860
|
| 37 |
|
modules/ai_model.py
CHANGED
|
@@ -22,7 +22,7 @@ class AIModel:
|
|
| 22 |
"""HuggingFace认证"""
|
| 23 |
try:
|
| 24 |
# 从环境变量获取HF token
|
| 25 |
-
hf_token = os.getenv("Assitant_tocken")
|
| 26 |
|
| 27 |
if hf_token:
|
| 28 |
login(token=hf_token, add_to_git_credential=False)
|
|
@@ -39,7 +39,7 @@ class AIModel:
|
|
| 39 |
try:
|
| 40 |
log.info(f"正在加载模型: {self.model_name}")
|
| 41 |
|
| 42 |
-
hf_token = os.getenv("Assitant_tocken")
|
| 43 |
|
| 44 |
self.model = Gemma3nForConditionalGeneration.from_pretrained(
|
| 45 |
self.model_name,
|
|
|
|
| 22 |
"""HuggingFace认证"""
|
| 23 |
try:
|
| 24 |
# 从环境变量获取HF token
|
| 25 |
+
hf_token = os.getenv("Assitant_tocken")
|
| 26 |
|
| 27 |
if hf_token:
|
| 28 |
login(token=hf_token, add_to_git_credential=False)
|
|
|
|
| 39 |
try:
|
| 40 |
log.info(f"正在加载模型: {self.model_name}")
|
| 41 |
|
| 42 |
+
hf_token = os.getenv("Assitant_tocken")
|
| 43 |
|
| 44 |
self.model = Gemma3nForConditionalGeneration.from_pretrained(
|
| 45 |
self.model_name,
|