Spaces:
Sleeping
Sleeping
Update src/backend/model_loader.py
Browse files
src/backend/model_loader.py
CHANGED
|
@@ -60,7 +60,7 @@ def load_model(
|
|
| 60 |
|
| 61 |
|
| 62 |
# Carrega tokenizer (pode ser de um reposit贸rio diferente)
|
| 63 |
-
tokenizer_model_name = args.
|
| 64 |
tokenizer = AutoTokenizer.from_pretrained( tokenizer_model_name )
|
| 65 |
|
| 66 |
# Adiciona pad_token se n茫o existir
|
|
|
|
| 60 |
|
| 61 |
|
| 62 |
# Carrega tokenizer (pode ser de um reposit贸rio diferente)
|
| 63 |
+
tokenizer_model_name = args.pop('tokenizer',model_name)
|
| 64 |
tokenizer = AutoTokenizer.from_pretrained( tokenizer_model_name )
|
| 65 |
|
| 66 |
# Adiciona pad_token se n茫o existir
|