From 3e237e9c82f82600943fe05a63234c0709915746 Mon Sep 17 00:00:00 2001 From: hailin Date: Fri, 8 Aug 2025 20:03:54 +0800 Subject: [PATCH] . --- check_train.sh | 3 +++ 1 file changed, 3 insertions(+) diff --git a/check_train.sh b/check_train.sh index 3cf983c..b3b8892 100644 --- a/check_train.sh +++ b/check_train.sh @@ -1,5 +1,8 @@ python - <<'PY' import warnings, torch +warnings.filterwarnings("ignore", category=UserWarning, message=".*TypedStorage is deprecated.*") +warnings.filterwarnings("ignore", category=UserWarning, message="Was asked to gather.*") + from datasets import load_dataset from transformers import ( AutoTokenizer, AutoModelForCausalLM,