Update README.md
Browse files
README.md
CHANGED
|
@@ -240,7 +240,7 @@ MODEL_DIR = "Harley-ml/Mini-MD"
|
|
| 240 |
TOKENIZER_DIR = MODEL_DIR
|
| 241 |
DEFAULT_MAX_NEW_TOKENS = 640
|
| 242 |
DEFAULT_TEMPERATURE = 0.9
|
| 243 |
-
DEFAULT_TOP_P =
|
| 244 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 245 |
PROMPT = ">>> "
|
| 246 |
# ---------------------------
|
|
|
|
| 240 |
TOKENIZER_DIR = MODEL_DIR
|
| 241 |
DEFAULT_MAX_NEW_TOKENS = 640
|
| 242 |
DEFAULT_TEMPERATURE = 0.9
|
| 243 |
+
DEFAULT_TOP_P = 1.4
|
| 244 |
DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
| 245 |
PROMPT = ">>> "
|
| 246 |
# ---------------------------
|