Spaces:
Build error
Build error
Update app.py
Browse files
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(
|