Create config.py

config.py

@@ -0,0 +1,15 @@
+# config.py
+
+import torch
+
+# Device configuration
+DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+
+# Path to the Yi-Coder model
+MODEL_PATH = "path_to_yi_coder_model"  # Replace with your actual model path
+
+# Maximum length for generated text
+MAX_LENGTH = 1024
+
+# Temperature for text generation
+TEMPERATURE = 0.7