Nac31 commited on
Commit
744b744
·
1 Parent(s): 3257b28

Update model used

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ hf_token = os.getenv('HF_TOKEN')
12
  login(hf_token)
13
 
14
  # Configuration du modèle
15
- model_path = "mistralai/Pixtral-Large-Instruct-2411"
16
  dtype = torch.bfloat16 if torch.cuda.get_device_capability()[0] == 8 else torch.float16
17
 
18
  # Initialisation du modèle
 
12
  login(hf_token)
13
 
14
  # Configuration du modèle
15
+ model_path = "mistralai/Mistral-Large-Instruct-2411"
16
  dtype = torch.bfloat16 if torch.cuda.get_device_capability()[0] == 8 else torch.float16
17
 
18
  # Initialisation du modèle