ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-09 20:31:12 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_diffusers.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-09-27 11:54:45 +02:00 committed by GitHub
parent dc144d6d9c
commit 242bdb68dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -240,13 +240,16 @@ class LollmsDiffusers(LollmsTTI):
height = adjust_dimensions(int(height)) height = adjust_dimensions(int(height))
if output_path is None: if output_path is None:
output_path = self.output_dir output_path = self.output_dir
output_path = Path(output_path)
fn = find_next_available_filename(output_path,"diff_img_")
else:
output_path = Path(output_path)
fn = output_path
if seed!=-1: if seed!=-1:
generator = torch.Generator("cuda").manual_seed(seed) generator = torch.Generator("cuda").manual_seed(seed)
image = self.model(positive_prompt, negative_prompt=negative_prompt, height=height, width=width, guidance_scale=scale, num_inference_steps=steps, generator=generator).images[0] image = self.model(positive_prompt, negative_prompt=negative_prompt, height=height, width=width, guidance_scale=scale, num_inference_steps=steps, generator=generator).images[0]
else: else:
image = self.model(positive_prompt, negative_prompt=negative_prompt, height=height, width=width, guidance_scale=scale, num_inference_steps=steps).images[0] image = self.model(positive_prompt, negative_prompt=negative_prompt, height=height, width=width, guidance_scale=scale, num_inference_steps=steps).images[0]
output_path = Path(output_path)
fn = find_next_available_filename(output_path,"diff_img_")
# Save the image # Save the image
image.save(fn) image.save(fn)
return fn, {"prompt":positive_prompt, "negative_prompt":negative_prompt} return fn, {"prompt":positive_prompt, "negative_prompt":negative_prompt}