mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
Update mtbench101.py (#1276)
fix wrong-used import from torch.utils.data import DataLoader, Dataset
This commit is contained in:
parent
26d077b080
commit
842fb1cd70
@ -5,7 +5,6 @@ import re
|
||||
from typing import Optional
|
||||
|
||||
from datasets import Dataset, DatasetDict
|
||||
from torch.utils.data import DataLoader, Dataset
|
||||
|
||||
from opencompass.registry import LOAD_DATASET
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user