Fix tests import

This commit is contained in:
Joshua Lochner 2023-05-31 18:48:06 +02:00
parent 861911fa12
commit 76dc236e3f
1 changed files with 1 additions and 1 deletions

View File

@ -6,7 +6,7 @@ import os
from transformers import AutoTokenizer, AutoConfig
from scripts.tasks import SUPPORTED_MODELS
from scripts.supported_models import SUPPORTED_MODELS
MODELS_TO_TEST = {
"albert": [