C2MV commited on
Commit
8a4a425
1 Parent(s): 10d490c

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -6
config.py CHANGED
@@ -2,14 +2,14 @@
2
 
3
  import torch
4
 
5
- # Device configuration
6
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
7
 
8
- # Path to the Yi-Coder model
9
- MODEL_PATH = "01-ai/Yi-Coder-9B-Chat" # Replace with your actual model path
10
 
11
- # Maximum length for generated text
12
  MAX_LENGTH = 1024
13
 
14
- # Temperature for text generation
15
- TEMPERATURE = 0.7
 
2
 
3
  import torch
4
 
5
+ # Configuración del dispositivo
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
13
 
14
+ # Temperatura para la generación de texto
15
+ TEMPERATURE = 0.7