mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
monkey patch transformers version checks
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
@@ -3,6 +3,7 @@ from functools import partial
|
||||
import os
|
||||
import re
|
||||
import sys
|
||||
import types
|
||||
import logging
|
||||
import warnings
|
||||
import urllib3
|
||||
@@ -133,6 +134,12 @@ timer.startup.record("accelerate")
|
||||
import pydantic # pylint: disable=W0611,C0411
|
||||
timer.startup.record("pydantic")
|
||||
|
||||
try:
|
||||
fake_version_check = types.ModuleType("transformers.dependency_versions_check")
|
||||
sys.modules["transformers.dependency_versions_check"] = fake_version_check # disable transformers version checks
|
||||
fake_version_check.dep_version_check = lambda pkg, hint=None: None
|
||||
except Exception:
|
||||
pass
|
||||
import transformers # pylint: disable=W0611,C0411
|
||||
from transformers import logging as transformers_logging # pylint: disable=W0611,C0411
|
||||
transformers_logging.set_verbosity_error()
|
||||
|
||||
Reference in New Issue
Block a user