From 4dc0e81c3d12f91ba379237afb7cc7c7e066ddfc Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Fri, 7 Jun 2024 23:45:59 +0200 Subject: [PATCH] upgraded diffusers --- lollms/services/diffusers/lollms_diffusers.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/lollms/services/diffusers/lollms_diffusers.py b/lollms/services/diffusers/lollms_diffusers.py index ff67fb7..db6bb4c 100644 --- a/lollms/services/diffusers/lollms_diffusers.py +++ b/lollms/services/diffusers/lollms_diffusers.py @@ -8,7 +8,7 @@ import sys from lollms.app import LollmsApplication from lollms.paths import LollmsPaths from lollms.config import TypedConfig, ConfigTemplate, BaseConfig -from lollms.utilities import PackageManager, check_and_install_torch, find_next_available_filename +from lollms.utilities import PackageManager, check_and_install_torch, find_next_available_filename, install_cuda, check_torch_version import time import io import sys @@ -122,7 +122,11 @@ class LollmsDiffusers(LollmsTTI): import torch if not PackageManager.check_package_installed("diffusers"): + check_and_install_torch("nvidia" in self.app.config.hardware_mode) PackageManager.install_or_update("diffusers") + PackageManager.install_or_update("sentencepiece") + PackageManager.install_or_update("accelerate") + from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image#PixArtSigmaPipeline self.model = AutoPipelineForText2Image.from_pretrained( app.config.diffusers_model, torch_dtype=torch.float16, cache_dir=self.models_dir,