1580158
/
multidiffusion-upscaler-for-automatic1111
mirror of https://github.com/pkuliyi2015/multidiffusion-upscaler-for-automatic1111.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

random seed across batch

pull/157/head
pkuliyi2015 2023-04-02 05:01:55 +00:00
parent 6f7cd4fed2
commit c404b4e036
2 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
javascript/bboxHint.js
View File

@ -476,7 +476,7 @@ function getSeedInfo(is_t2i, id, current_seed){
info = info.replace(/<[^>]*>/g, '');
// Find a json string 'region control:' in the info
// get its index
idx = info.indexOf('region control');
idx = info.indexOf('Region control');
if (idx == -1) return current_seed;
// get the json string (detect the bracket)
// find the first '{'
@ -500,7 +500,7 @@ function getSeedInfo(is_t2i, id, current_seed){
// parse the json string
let json = JSON.parse(json_str);
// get the seed if the region id is in the json
const region_id = 'region ' + id.toString();
const region_id = 'Region ' + id.toString();
if (!(region_id in json)) return current_seed;
const region = json[region_id];
if (!('seed' in region)) return current_seed;

19
scripts/tilediffusion.py
View File

@ -343,9 +343,10 @@ class Script(scripts.Script):
info['Keep input size'] = keep_input_size
if noise_inverse:
info['Noise inverse'] = True
info['steps'] = noise_inverse_steps
info['renoise strength'] = noise_inverse_renoise_strength
info['kernel size'] = noise_inverse_renoise_kernel
info['Steps'] = noise_inverse_steps
info['Retouch'] = noise_inverse_retouch
info['Renoise strength'] = noise_inverse_renoise_strength
info['Kernel size'] = noise_inverse_renoise_kernel
if enable_bbox_control:
@ -354,8 +355,8 @@ class Script(scripts.Script):
region_settings = {}
for i, v in bbox_settings.items():
region_settings['region ' + str(i+1)] = v._asdict()
info["region control"] = region_settings
region_settings['Region ' + str(i+1)] = v._asdict()
info["Region control"] = region_settings
def create_bbox_random_tensors(shape, seeds, subseeds=None, subseed_strength=0.0, seed_resize_from_h=0, seed_resize_from_w=0, p=None):
org_random_tensors = processing.create_random_tensors_original_md(shape, seeds, subseeds, subseed_strength, seed_resize_from_h, seed_resize_from_w, p)
@ -365,7 +366,7 @@ class Script(scripts.Script):
foreground_noise = torch.zeros_like(org_random_tensors)
foreground_noise_count = torch.zeros((1, 1, height, width), device=org_random_tensors.device)
for i, v in bbox_settings.items():
seed = region_settings['region ' + str(i+1)]['seed']
seed = v.seed
if seed == -1 or seed == '':
seed = int(random.randrange(4294967294))
x, y, w, h = v.x, v.y, v.w, v.h
@ -390,10 +391,8 @@ class Script(scripts.Script):
foreground_noise_count[:, :, y:y+h, x:x+w] += 1
else:
raise NotImplementedError
# update seed for the next batch
region_settings['region ' + str(i+1)]['seed'] = seed
seed += 1
bbox_settings[i] = v._replace(seed=seed)
# update seed in the PNG info
region_settings['Region ' + str(i+1)]['seed'] = seed
# average
background_noise = torch.where(background_noise_count > 1, background_noise / background_noise_count, background_noise)