Fix tests import
This commit is contained in:
parent
861911fa12
commit
76dc236e3f
|
@ -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": [
|
||||
|
|
Loading…
Reference in New Issue