C2MV commited on
Commit
28dea9b
1 Parent(s): e2c6e1c

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -6,7 +6,7 @@ import torch
6
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
  # Ruta al modelo Yi-Coder
9
- MODEL_PATH = "ruta_al_modelo_yi_coder" # Reemplaza con la ruta real de tu modelo
10
 
11
  # Longitud máxima para el texto generado
12
  MAX_LENGTH = 1024
 
6
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
  # Ruta al modelo Yi-Coder
9
+ MODEL_PATH = "01-ai/Yi-Coder-9B-Chat" # Reemplaza con la ruta real de tu modelo
10
 
11
  # Longitud máxima para el texto generado
12
  MAX_LENGTH = 1024