Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-08-14 16:57:20 -04:00
parent 001716633d
commit 1ce9aae143
2 changed files with 4 additions and 3 deletions
+2 -3
View File
@@ -199,8 +199,8 @@ def deprecate_warn(*args, **kwargs):
diffusers.utils.deprecation_utils.deprecate = deprecate_warn
diffusers.utils.deprecate = deprecate_warn
def patch_torch_version():
import torch
if not hasattr(torch, '__version_backup__'):
torch.__version_backup__ = torch.__version__
# Convert string version to tuple format to solve TypeError caused by BnB
@@ -215,8 +215,7 @@ def patch_torch_version():
return super().__ge__(other)
torch.__version__ = VersionString(torch.__version__)
patch_torch_version()
errors.log.info(f'Torch: torch=={torch.__version__} torchvision=={torchvision.__version__}')
errors.log.info(f'Packages: diffusers=={diffusers.__version__} transformers=={transformers.__version__} accelerate=={accelerate.__version__} gradio=={gradio.__version__} pydantic=={pydantic.__version__} numpy=={np.__version__}')
+2
View File
@@ -74,6 +74,8 @@ def copy_diffuser_options(new_pipe, orig_pipe):
def set_huggingface_options(op: str, model_type: str):
if model_type is not None: # overrides
pass
if shared.opts.diffusers_to_gpu: # and model_type.startswith('Stable Diffusion'):
shared.log.debug(f'Setting {op}: component=accelerate direct={shared.opts.diffusers_to_gpu}')
sd_hijack_accelerate.hijack_accelerate()