ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-09 03:44:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_diffusers.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-10-09 17:19:08 +02:00 committed by GitHub
parent 98d8d210df
commit 08f2b466c9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lollms/services/tti/diffusers/lollms_diffusers.py
View File

@ -128,7 +128,7 @@ class LollmsDiffusers(LollmsTTI):
self.output_dir.mkdir(parents=True, exist_ok=True)
self.models_dir.mkdir(parents=True, exist_ok=True)
ASCIIColors.red("")
ASCIIColors.red(" _ _ _ _ _ __ __ ")
ASCIIColors.red(" | | | | | | (_)/ _|/ _| ")
ASCIIColors.red(" | | ___ | | |_ __ ___ ___ __| |_| |_| |_ _ _ ___ ___ _ __ ___ ")
@ -137,7 +137,8 @@ class LollmsDiffusers(LollmsTTI):
ASCIIColors.red(" |______\___/|_|_|_| |_| |_|___/ \__,_|_|_| |_| \__,_|___/\___|_| |___/ ")
ASCIIColors.red(" ______ ")
ASCIIColors.red(" |______| ")
ASCIIColors.red("")
ASCIIColors.yellow(f"Using model: {app.config.diffusers_model}")
import torch
if not PackageManager.check_package_installed("diffusers"):
check_and_install_torch("nvidia" in self.app.config.hardware_mode)