prelington commited on
Commit
9c56b78
·
verified ·
1 Parent(s): 12ba2d2

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +6 -0
config.py ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ # config.py
2
+
3
+ MODEL_NAME = "your-username/ChatGpt-Tune" # Hugging Face model
4
+ MAX_TOKENS = 200
5
+ TEMPERATURE = 0.7
6
+ DEVICE = "cuda" # or "cpu"