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

Merge pull request #4233 from rabanti-github/dev

Browse Source 
Added option to randomize loopback iterations
This commit is contained in:
Vladimir Mandic
2025-09-28 09:06:20 -04:00
committed by GitHub
parent 4a70e82b0c 862e601906
commit d45b6dad40
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/loopback.py
+9 -2
View File
@@ -1,4 +1,5 @@
import math
import random
import gradio as gr
from modules import images, processing, scripts_manager
@@ -21,10 +22,12 @@ class Script(scripts_manager.Script):
final_denoising_strength = gr.Slider(minimum=0, maximum=1, step=0.01, label='Final strength', value=0.5, elem_id=self.elem_id("final_denoising_strength"))
with gr.Row():
denoising_curve = gr.Dropdown(label="Strength curve", choices=["Aggressive", "Linear", "Lazy"], value="Linear")
with gr.Row():
randomize_seed = gr.Checkbox(label="Randomize seed after each loop iteration", value=False)
return [loops, final_denoising_strength, denoising_curve]
return [loops, final_denoising_strength, denoising_curve, randomize_seed]
def run(self, p, loops, final_denoising_strength, denoising_curve): # pylint: disable=arguments-differ
def run(self, p, loops, final_denoising_strength, denoising_curve, randomize_seed): # pylint: disable=arguments-differ
processing.fix_seed(p)
initial_batch_count = p.n_iter
p.extra_generation_params['Loopback'] = final_denoising_strength
@@ -76,6 +79,10 @@ class Script(scripts_manager.Script):
if initial_seed is None:
initial_seed = processed.seed
initial_info = processed.info
if randomize_seed:
p.seed = random.randrange(4294967294)
p.all_seeds = [p.seed]
log.info(f'Setting random seed {p.seed} for loopback iteration {i}')
p.seed = processed.seed + 1 # why?
p.denoising_strength = calculate_denoising_strength(i + 1)
last_image = processed.images[0]