Fazzioni commited on
Commit
268ccb1
verified
1 Parent(s): 4fcddb4

Update src/backend/model_loader.py

Browse files
Files changed (1) hide show
  1. src/backend/model_loader.py +1 -1
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.get('tokenizer',model_name)
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