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

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-17 19:55:45 +02:00
parent 304035a669
commit 25c9ee57da
1 changed files with 27 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
lollms/services/sd/lollms_sd.py
View File

@ -372,62 +372,33 @@ class LollmsSD(LollmsTTI):
output_path = self.output_dir
infos = {}
img_paths = []
if len(files)>0:
try:
generated = self.img2img(
positive_prompt,
negative_prompt,
[self.loadImage(files[-1])],
sampler_name=sampler_name,
seed=seed,
cfg_scale=scale,
steps=steps,
width=int(width),
height=int(height),
denoising_strength=img2img_denoising_strength,
tiling=False,
restore_faces=restore_faces,
styles=None,
script_name="",
)
"""
images: list
parameters: dict
info: dict
"""
for img in generated.images:
img_paths.append(self.saveImage(img, output_path))
infos = generated.info
except Exception as ex:
ASCIIColors.error("Couldn't generate the image")
trace_exception(ex)
else:
try:
generated = self.txt2img(
positive_prompt,
negative_prompt=negative_prompt,
sampler_name=sampler_name,
seed=seed,
cfg_scale=scale,
steps=steps,
width=width,
height=height,
tiling=False,
restore_faces=restore_faces,
styles=None,
script_name="",
)
"""
images: list
parameters: dict
info: dict
"""
for img in generated.images:
img_paths.append(self.saveImage(img, output_path))
infos = generated.info
except Exception as ex:
ASCIIColors.error("Couldn't generate the image")
trace_exception(ex)
try:
generated = self.txt2img(
positive_prompt,
negative_prompt=negative_prompt,
sampler_name=sampler_name,
seed=seed,
cfg_scale=scale,
steps=steps,
width=width,
height=height,
tiling=False,
restore_faces=restore_faces,
styles=None,
script_name="",
)
"""
images: list
parameters: dict
info: dict
"""
for img in generated.images:
img_paths.append(self.saveImage(img, output_path))
infos = generated.info
except Exception as ex:
ASCIIColors.error("Couldn't generate the image")
trace_exception(ex)
return img_paths[0] if len(img_paths)>0 else None, infos