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

fix: 修复hf_tocken

Browse files
Files changed (1) hide show
  1. modules/ai_model.py +2 -2
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("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") or 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") or os.getenv("gemma3n_git")
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") or os.getenv("gemma3n_git")
43
 
44
  self.model = Gemma3nForConditionalGeneration.from_pretrained(
45
  self.model_name,