Eliot0110 commited on
Commit
3154fce
·
1 Parent(s): 82e8be7

improve: simplify tocken use

Browse files
Files changed (1) hide show
  1. modules/ai_model.py +1 -0
modules/ai_model.py CHANGED
@@ -40,6 +40,7 @@ class AIModel:
40
  cache_dir = "/app/.cache/huggingface"
41
  login(token=token_to_use, add_to_git_credential=False)
42
  log.info("✅ HuggingFace 认证成功")
 
43
 
44
 
45
 
 
40
  cache_dir = "/app/.cache/huggingface"
41
  login(token=token_to_use, add_to_git_credential=False)
42
  log.info("✅ HuggingFace 认证成功")
43
+ return token_to_use
44
 
45
 
46