C2MV commited on
Commit
9990e20
1 Parent(s): 17a3d9f

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 = "cuda" if torch.cuda.is_available() else "cpu"
7
 
8
  # Path to the Yi-Coder model
9
- MODEL_PATH = "path_to_yi_coder_model" # Replace with your actual model path
10
 
11
  # Maximum length for generated text
12
  MAX_LENGTH = 1024
 
6
  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