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
# All paths are set with the intent you should run this script from the root of the repo with the command
@@ -30,7 +31,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"]
CONFIG_MAPPING = transformers.models.auto.configuration_auto.CONFIG_MAPPING