Update deprecated load_module (#21651)

This commit is contained in:
Sylvain Gugger
2023-02-15 15:57:24 -05:00
committed by GitHub
parent 1567bef3b3
commit 9d1116e995
10 changed files with 39 additions and 10 deletions

View File

@@ -17,6 +17,7 @@ import importlib
import inspect
import os
import re
import sys
import warnings
from collections import OrderedDict
from difflib import get_close_matches
@@ -311,7 +312,9 @@ spec = importlib.util.spec_from_file_location(
os.path.join(PATH_TO_TRANSFORMERS, "__init__.py"),
submodule_search_locations=[PATH_TO_TRANSFORMERS],
)
transformers = spec.loader.load_module()
transformers = importlib.util.module_from_spec(spec)
spec.loader.exec_module(transformers)
transformers = sys.modules["transformers"]
def check_model_list():