Spaces:
Running
Running
fix: 修复hf_tocken
Browse files- 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("
|
| 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("
|
| 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,
|