Spaces:

C2MV
/

Biotech2

Running on Zero

C2MV commited on Sep 21

Commit

9d0c2b1

•

1 Parent(s): 778ff50

Update config.py

Files changed (1) hide show

config.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torch
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Ruta al modelo Yi-Coder
-MODEL_PATH = "01-ai/Yi-Coder-9B-Chat"  # Reemplaza con la ruta real de tu modelo
 # Longitud máxima para el texto generado
 MAX_LENGTH = 1024

 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Ruta al modelo Yi-Coder
+MODEL_PATH = "01-ai/Yi-Coder-1.5B-Chat"  # Reemplaza con la ruta real de tu modelo
 # Longitud máxima para el texto generado
 MAX_LENGTH = 1024