From 43247bf0f58d57dd8aa9824a8cd954901117500e Mon Sep 17 00:00:00 2001 From: n1ck-guo Date: Fri, 17 Oct 2025 00:16:40 -0400 Subject: [PATCH] block dataset logger info Signed-off-by: n1ck-guo --- auto_round/calib_dataset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/auto_round/calib_dataset.py b/auto_round/calib_dataset.py index 4825177b6..5bac8a419 100644 --- a/auto_round/calib_dataset.py +++ b/auto_round/calib_dataset.py @@ -13,9 +13,12 @@ # limitations under the License. import json +import logging import random import sys +logging.getLogger("datasets").setLevel(logging.WARNING) + import torch from datasets import Dataset, Features, IterableDataset, Sequence, Value, concatenate_datasets, load_dataset from torch.utils.data import DataLoader