remove browser to extension
This commit is contained in:
parent
124e44cf1e
commit
cef1b89aa2
4 changed files with 3 additions and 59 deletions
|
@ -1,3 +1,4 @@
|
||||||
|
|
||||||
callbacks_model_loaded = []
|
callbacks_model_loaded = []
|
||||||
callbacks_ui_tabs = []
|
callbacks_ui_tabs = []
|
||||||
callbacks_ui_settings = []
|
callbacks_ui_settings = []
|
||||||
|
@ -15,6 +16,7 @@ def model_loaded_callback(sd_model):
|
||||||
|
|
||||||
def ui_tabs_callback():
|
def ui_tabs_callback():
|
||||||
res = []
|
res = []
|
||||||
|
|
||||||
for callback in callbacks_ui_tabs:
|
for callback in callbacks_ui_tabs:
|
||||||
res += callback() or []
|
res += callback() or []
|
||||||
|
|
||||||
|
|
|
@ -82,7 +82,6 @@ parser.add_argument("--api", action='store_true', help="use api=True to launch t
|
||||||
parser.add_argument("--nowebui", action='store_true', help="use api=True to launch the api instead of the webui")
|
parser.add_argument("--nowebui", action='store_true', help="use api=True to launch the api instead of the webui")
|
||||||
parser.add_argument("--ui-debug-mode", action='store_true', help="Don't load model to quickly launch UI")
|
parser.add_argument("--ui-debug-mode", action='store_true', help="Don't load model to quickly launch UI")
|
||||||
parser.add_argument("--device-id", type=str, help="Select the default CUDA device to use (export CUDA_VISIBLE_DEVICES=0,1,etc might be needed before)", default=None)
|
parser.add_argument("--device-id", type=str, help="Select the default CUDA device to use (export CUDA_VISIBLE_DEVICES=0,1,etc might be needed before)", default=None)
|
||||||
parser.add_argument("--browse-all-images", action='store_true', help="Allow browsing all images by Image Browser", default=False)
|
|
||||||
|
|
||||||
cmd_opts = parser.parse_args()
|
cmd_opts = parser.parse_args()
|
||||||
restricted_opts = [
|
restricted_opts = [
|
||||||
|
|
|
@ -1104,7 +1104,7 @@ def create_ui(wrap_gradio_gpu_call):
|
||||||
upscaling_crop = gr.Checkbox(label='Crop to fit', value=True)
|
upscaling_crop = gr.Checkbox(label='Crop to fit', value=True)
|
||||||
|
|
||||||
with gr.Group():
|
with gr.Group():
|
||||||
extras_upscaler_1 = gr.Radio(label='Upscaler 1', elem_id="extras_upscaler_1", choices=[x.name for x in shared.sd_upscalers] , value=shared.sd_upscalers[0].name, type="index")
|
extras_upscaler_1 = gr.Radio(label='Upscaler 1', elem_id="extras_upscaler_1", choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name, type="index")
|
||||||
|
|
||||||
with gr.Group():
|
with gr.Group():
|
||||||
extras_upscaler_2 = gr.Radio(label='Upscaler 2', elem_id="extras_upscaler_2", choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name, type="index")
|
extras_upscaler_2 = gr.Radio(label='Upscaler 2', elem_id="extras_upscaler_2", choices=[x.name for x in shared.sd_upscalers], value=shared.sd_upscalers[0].name, type="index")
|
||||||
|
|
|
@ -1,57 +0,0 @@
|
||||||
import csv, os, shutil
|
|
||||||
import modules.scripts as scripts
|
|
||||||
from modules import processing, shared, sd_samplers, images
|
|
||||||
from modules.processing import Processed
|
|
||||||
from modules.shared import opts
|
|
||||||
import gradio
|
|
||||||
class Script(scripts.Script):
|
|
||||||
def title(self):
|
|
||||||
return "Create inspiration images"
|
|
||||||
|
|
||||||
def show(self, is_img2img):
|
|
||||||
return True
|
|
||||||
|
|
||||||
def ui(self, is_img2img):
|
|
||||||
file = gradio.Files(label="Artist or styles name list. '.txt' files with one name per line",)
|
|
||||||
with gradio.Row():
|
|
||||||
prefix = gradio.Textbox("a painting in", label="Prompt words before artist or style name", file_count="multiple")
|
|
||||||
suffix= gradio.Textbox("style", label="Prompt words after artist or style name")
|
|
||||||
negative_prompt = gradio.Textbox("picture frame, portrait photo", label="Negative Prompt")
|
|
||||||
with gradio.Row():
|
|
||||||
batch_size = gradio.Number(1, label="Batch size")
|
|
||||||
batch_count = gradio.Number(2, label="Batch count")
|
|
||||||
return [batch_size, batch_count, prefix, suffix, negative_prompt, file]
|
|
||||||
|
|
||||||
def run(self, p, batch_size, batch_count, prefix, suffix, negative_prompt, files):
|
|
||||||
p.batch_size = int(batch_size)
|
|
||||||
p.n_iterint = int(batch_count)
|
|
||||||
p.negative_prompt = negative_prompt
|
|
||||||
p.do_not_save_samples = True
|
|
||||||
p.do_not_save_grid = True
|
|
||||||
for file in files:
|
|
||||||
tp = file.orig_name.split(".")[0]
|
|
||||||
print(tp)
|
|
||||||
path = os.path.join(opts.inspiration_dir, tp)
|
|
||||||
if not os.path.exists(path):
|
|
||||||
os.makedirs(path)
|
|
||||||
f = open(file.name, "r")
|
|
||||||
line = f.readline()
|
|
||||||
while len(line) > 0:
|
|
||||||
name = line.rstrip("\n").split(",")[0]
|
|
||||||
line = f.readline()
|
|
||||||
artist_path = os.path.join(path, name)
|
|
||||||
if not os.path.exists(artist_path):
|
|
||||||
os.mkdir(artist_path)
|
|
||||||
if len(os.listdir(artist_path)) >= opts.inspiration_max_samples:
|
|
||||||
continue
|
|
||||||
p.prompt = f"{prefix} {name} {suffix}"
|
|
||||||
print(p.prompt)
|
|
||||||
processed = processing.process_images(p)
|
|
||||||
for img in processed.images:
|
|
||||||
i = 0
|
|
||||||
filename = os.path.join(artist_path, format(0, "03d") + ".jpg")
|
|
||||||
while os.path.exists(filename):
|
|
||||||
i += 1
|
|
||||||
filename = os.path.join(artist_path, format(i, "03d") + ".jpg")
|
|
||||||
img.save(filename, quality=80)
|
|
||||||
return processed
|
|
Loading…
Reference in a new issue