diff options
| author | blackhao <13851610112@163.com> | 2025-08-23 15:03:34 -0500 |
|---|---|---|
| committer | blackhao <13851610112@163.com> | 2025-08-23 15:03:34 -0500 |
| commit | 5539e387c08a54c77b9cb5e92faf7d878bff40ad (patch) | |
| tree | 3ccca5b8acc6873a4a589eac4e82f1de3f78e181 /Group-Entropy-Equalization | |
| parent | 8b2cf4b226de17227aa95abb637d1410bc2e57e3 (diff) | |
maybe
Diffstat (limited to 'Group-Entropy-Equalization')
| -rw-r--r-- | Group-Entropy-Equalization/train.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/Group-Entropy-Equalization/train.py b/Group-Entropy-Equalization/train.py index 9ca1d4a..352e916 100644 --- a/Group-Entropy-Equalization/train.py +++ b/Group-Entropy-Equalization/train.py @@ -18,6 +18,7 @@ from accelerate import Accelerator, DeepSpeedPlugin from accelerate.utils import set_seed from transformers import AutoConfig, AutoTokenizer, AutoModelForCausalLM import json +import math os.environ.setdefault("NCCL_TIMEOUT", "2700") |