patch diffusers loader

This commit is contained in:
Vladimir Mandic
2024-05-10 10:08:01 -04:00
parent e7b30ef662
commit 2f3557f6ea
3 changed files with 6 additions and 2 deletions
+5
View File
@@ -4,6 +4,7 @@ import sys
import logging
import warnings
import urllib3
from functools import partial
from modules import timer, errors
initialized = False
@@ -53,6 +54,10 @@ import pydantic # pylint: disable=W0611,C0411
timer.startup.record("pydantic")
import diffusers # pylint: disable=W0611,C0411
import diffusers.loaders.single_file # pylint: disable=W0611,C0411
logging.getLogger("diffusers.loaders.single_file").setLevel(logging.ERROR)
from tqdm.rich import tqdm
diffusers.loaders.single_file.logging.tqdm = partial(tqdm, unit='C')
timer.startup.record("diffusers")
def get_packages():