Update README.en.md
Browse files- README.en.md +2 -2
README.en.md
CHANGED
@@ -17,10 +17,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
17 |
import torch
|
18 |
import os
|
19 |
|
20 |
-
# Adjustable parameters; it is recommended to set them to higher values during text generation
|
21 |
TOP_P = 0.9 # Top-p (nucleus sampling), range from 0 to 1
|
22 |
TOP_K = 80 # Top-k sampling value K
|
23 |
-
TEMPERATURE = 0.
|
24 |
|
25 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
26 |
|
|
|
17 |
import torch
|
18 |
import os
|
19 |
|
20 |
+
# Adjustable parameters; it is recommended to set them to higher values during text generation (Except Temperature)
|
21 |
TOP_P = 0.9 # Top-p (nucleus sampling), range from 0 to 1
|
22 |
TOP_K = 80 # Top-k sampling value K
|
23 |
+
TEMPERATURE = 0.3 # Temperature parameter to control randomness in text generation
|
24 |
|
25 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
26 |
|