baconnier commited on
Commit
213cdcd
·
verified ·
1 Parent(s): 8075ed3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -58,6 +58,8 @@ config = AutoConfig.from_pretrained(model_id)
58
  # Ajouter l'attribut vocab_size manuellement si nécessaire
59
  if not hasattr(config, "vocab_size"):
60
  config.vocab_size = 32000 # Valeur standard pour les modèles basés sur Llama
 
 
61
 
62
  tokenizer = AutoTokenizer.from_pretrained(model_id)
63
  model = AutoModelForCausalLM.from_pretrained(
 
58
  # Ajouter l'attribut vocab_size manuellement si nécessaire
59
  if not hasattr(config, "vocab_size"):
60
  config.vocab_size = 32000 # Valeur standard pour les modèles basés sur Llama
61
+ if not hasattr(config, "hidden_size"):
62
+ config.hidden_size = 2560 # Valeur standard pour les modèles 4B
63
 
64
  tokenizer = AutoTokenizer.from_pretrained(model_id)
65
  model = AutoModelForCausalLM.from_pretrained(