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,