Spaces:
Running
on
Zero
Running
on
Zero
Upload step03_chatbot.py with huggingface_hub
Browse files- 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...")
|