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