1580158
/
SD-latent-mirroring
mirror of https://github.com/dfaker/SD-latent-mirroring.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Delete latent_mirroring_always_on.py

pull/6/head
DepFA 2022-11-06 00:48:12 +00:00 committed by GitHub
parent 2e632a362e
commit 94b9cc79d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 60 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
latent_mirroring_always_on.py
View File

@ -1,60 +0,0 @@
import torch
import modules.scripts as scripts
import gradio as gr
from modules.script_callbacks import on_cfg_denoiser
from modules import processing
class Script(scripts.Script):
def title(self):
return "Latent Mirroring"
def show(self, is_img2img):
return scripts.AlwaysVisible
def ui(self, is_img2img):
mirror_mode = gr.Radio(label='Mirror application mode', choices=['None', 'Alternate Steps', 'Blend Average'], value='Alternate Steps', type="index")
mirror_style = gr.Radio(label='Mirror style', choices=['Vertical Mirroring', 'Horizontal Mirroring', '90 Degree Rotation', '180 Degree Rotation'], value='Vertical Mirroring', type="index")
mirroring_max_step_fraction = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label='Maximum steps fraction to mirror at', value=0.25)
self.run_callback = False
return [mirror_mode, mirror_style, mirroring_max_step_fraction]
def denoise_callback(self, params):
if self.run_callback and params.sampling_step < params.total_sampling_steps*self.mirroring_max_step_fraction:
if self.mirror_mode == 1:
if self.mirror_style == 0:
params.x[:, :, :, :] = torch.flip(params.x, [3])
elif self.mirror_style == 1:
params.x[:, :, :, :] = torch.flip(params.x, [2])
elif self.mirror_style == 2:
params.x[:, :, :, :] = torch.rot90(params.x, dims=[2, 3])
elif self.mirror_style == 3:
params.x[:, :, :, :] = torch.rot90(torch.rot90(params.x, dims=[2, 3]), dims=[2, 3])
elif self.mirror_mode == 2:
if self.mirror_style == 0:
params.x[:, :, :, :] = (torch.flip(params.x, [3]) + params.x)/2
elif self.mirror_style == 1:
params.x[:, :, :, :] = (torch.flip(params.x, [2]) + params.x)/2
elif self.mirror_style == 2:
params.x[:, :, :, :] = (torch.rot90(params.x, dims=[2, 3]) + params.x)/2
elif self.mirror_style == 3:
params.x[:, :, :, :] = (torch.rot90(torch.rot90(params.x_in, dims=[2, 3]), dims=[2, 3]) + params.x_in)/2
def process(self, p, mirror_mode, mirror_style, mirroring_max_step_fraction):
self.mirror_mode = mirror_mode
self.mirror_style = mirror_style
self.mirroring_max_step_fraction = mirroring_max_step_fraction
if not hasattr(self, 'callbacks_added'):
on_cfg_denoiser(self.denoise_callback)
self.callbacks_added = True
self.run_callback = True
def postprocess(self, *args):
self.run_callback = False
return result