ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-01 08:48:19 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded diffusers

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-07 23:45:59 +02:00
parent 731c418d1c
commit 4dc0e81c3d
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lollms/services/diffusers/lollms_diffusers.py
View File

@ -8,7 +8,7 @@ import sys
from lollms.app import LollmsApplication from lollms.app import LollmsApplication
from lollms.paths import LollmsPaths from lollms.paths import LollmsPaths
from lollms.config import TypedConfig, ConfigTemplate, BaseConfig 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 time
import io import io
import sys import sys
@ -122,7 +122,11 @@ class LollmsDiffusers(LollmsTTI):
import torch import torch
if not PackageManager.check_package_installed("diffusers"): 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("diffusers")
PackageManager.install_or_update("sentencepiece")
PackageManager.install_or_update("accelerate")
from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image#PixArtSigmaPipeline from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image#PixArtSigmaPipeline
self.model = AutoPipelineForText2Image.from_pretrained( self.model = AutoPipelineForText2Image.from_pretrained(
app.config.diffusers_model, torch_dtype=torch.float16, cache_dir=self.models_dir, app.config.diffusers_model, torch_dtype=torch.float16, cache_dir=self.models_dir,