__init__.py 198 Bytes
Newer Older
luopl's avatar
luopl committed
1
2
3
4
5
6
7
import os

os.environ["TOKENIZERS_PARALLELISM"] = "false"

# for torch==2.4.0
import warnings
warnings.filterwarnings("ignore", category=FutureWarning, module="torch.utils.checkpoint", lineno=1399)