Eliot0110 commited on
Commit
8fa71e2
·
1 Parent(s): cb807de

fix: 修复CLI认证环境变量名称

Browse files

- 统一使用 Assitant_tocken 作为环境变量名
- 在Dockerfile构建时进行HF认证
- 确保所有组件使用相同的变量名

Files changed (2) hide show
  1. Dockerfile +6 -0
  2. 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") or os.getenv("gemma3n_git")
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") or os.getenv("gemma3n_git")
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,