VincentGOURBIN commited on
Commit
9f90fad
·
verified ·
1 Parent(s): c9f3186

Upload step03_chatbot.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. step03_chatbot.py +2 -1
step03_chatbot.py CHANGED
@@ -197,10 +197,11 @@ class Qwen3Reranker:
197
 
198
  # Chargement du tokenizer
199
  print(" - Chargement du tokenizer...")
200
- self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
201
 
202
  # Configuration du modèle selon la plateforme
203
  model_kwargs = self._get_model_config()
 
204
 
205
  # Chargement du modèle
206
  print(" - Chargement du modèle...")
 
197
 
198
  # Chargement du tokenizer
199
  print(" - Chargement du tokenizer...")
200
+ self.tokenizer = AutoTokenizer.from_pretrained(self.model_name, trust_remote_code=True)
201
 
202
  # Configuration du modèle selon la plateforme
203
  model_kwargs = self._get_model_config()
204
+ model_kwargs['trust_remote_code'] = True # Requis pour Qwen3-Reranker
205
 
206
  # Chargement du modèle
207
  print(" - Chargement du modèle...")