Eliot0110 commited on
Commit
2bef76a
·
1 Parent(s): aa5bcf3

fix: use right tocken path

Browse files
Files changed (1) hide show
  1. modules/ai_model.py +2 -4
modules/ai_model.py CHANGED
@@ -87,16 +87,14 @@ class AIModel:
87
  torch_dtype=torch.bfloat16,
88
  trust_remote_code=True,
89
  token=token,
90
- cache_dir=cache_dir, # 明确指定缓存目录
91
- use_auth_token=token
92
  ).eval()
93
 
94
  self.processor = AutoProcessor.from_pretrained(
95
  self.model_name,
96
  trust_remote_code=True,
97
  token=token,
98
- cache_dir=cache_dir, # 明确指定缓存目录
99
- use_auth_token=token
100
  )
101
 
102
  log.info("✅ Gemma AI 模型初始化成功")
 
87
  torch_dtype=torch.bfloat16,
88
  trust_remote_code=True,
89
  token=token,
90
+ cache_dir=cache_dir
 
91
  ).eval()
92
 
93
  self.processor = AutoProcessor.from_pretrained(
94
  self.model_name,
95
  trust_remote_code=True,
96
  token=token,
97
+ cache_dir=cache_dir
 
98
  )
99
 
100
  log.info("✅ Gemma AI 模型初始化成功")