Hugging Face's logo

Spaces:
jojonocode
/
transw
Sleeping

App Files Community
jojonocode commited on
Commit
40f9d6c
·
verified ·
1 Parent(s): bc1d457

remaj model param

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
5
  # --------------------------------------------------
6
  # Chargement du modèle NLLB
7
  # --------------------------------------------------
8
- MODEL_NAME = "facebook/nllb-200-distilled-1.3B"
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  print(f"🚀 Chargement du modèle {MODEL_NAME} sur {device}...")
 
5
  # --------------------------------------------------
6
  # Chargement du modèle NLLB
7
  # --------------------------------------------------
8
+ MODEL_NAME = "facebook/nllb-200-3.3B"
9
 
10
  device = "cuda" if torch.cuda.is_available() else "cpu"
11
  print(f"🚀 Chargement du modèle {MODEL_NAME} sur {device}...")