Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-26 15:16:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

ONNX Stable Diffusion Inpaint WIP

Browse Source 
ONNX Stable Diffusion XL Img2Img WIP
ONNX Stable Diffusion XL Refiner WIP
and lots of fixes.
This commit is contained in:
Seunghoon Lee
2024-01-01 06:05:46 +09:00
parent 364c3beb2e
commit 3fb21306f6
6 changed files with 177 additions and 409 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+3 -1
View File
@@ -67,7 +67,9 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
for j in range(len(decoded)):
images.save_image(decoded[j], path=p.outpath_samples, basename="", seed=p.seeds[i], prompt=p.prompts[i], extension=shared.opts.samples_format, info=info, p=p, suffix=suffix)
def diffusers_callback_legacy(step: int, timestep: int, latents: torch.FloatTensor):
def diffusers_callback_legacy(step: int, timestep: int, latents: typing.Union[torch.FloatTensor, np.ndarray]):
if isinstance(latents, np.ndarray): # latents from Onnx pipelines is ndarray.
latents = torch.from_numpy(latents)
shared.state.sampling_step = step
shared.state.current_latent = latents
latents = processing_correction.correction_callback(p, timestep, {'latents': latents})