aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authoryfszzx <yfszzx@gmail.com>2022-10-24 10:10:33 +0800
committeryfszzx <yfszzx@gmail.com>2022-10-24 10:10:33 +0800
commitcef1b89aa2e6c7647db7e93a4cd4ec020da3f2da (patch)
treec41f530c47866587e0e0aab201e77227a0d987cf
parent124e44cf1eed1edc68954f63a2a9bc428aabbcec (diff)
remove browser to extension
-rw-r--r--modules/script_callbacks.py2
-rw-r--r--modules/shared.py1
-rw-r--r--modules/ui.py2
-rw-r--r--scripts/create_inspiration_images.py57
4 files changed, 3 insertions, 59 deletions
diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index 66666a56..f46d3d9a 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -1,3 +1,4 @@
+
callbacks_model_loaded = []
callbacks_ui_tabs = []
callbacks_ui_settings = []
@@ -15,6 +16,7 @@ def model_loaded_callback(sd_model):
def ui_tabs_callback():
res = []
+
for callback in callbacks_ui_tabs:
res += callback() or []
diff --git a/modules/shared.py b/modules/shared.py
index 5dfd7927..6541e679 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -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("--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("--browse-all-images", action='store_true', help="Allow browsing all images by Image Browser", default=False)
cmd_opts = parser.parse_args()
restricted_opts = [
diff --git a/modules/ui.py b/modules/ui.py
index fa42712e..a32f7259 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -1104,7 +1104,7 @@ def create_ui(wrap_gradio_gpu_call):
upscaling_crop = gr.Checkbox(label='Crop to fit', value=True)
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():
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")
diff --git a/scripts/create_inspiration_images.py b/scripts/create_inspiration_images.py
deleted file mode 100644
index 2fd30578..00000000
--- a/scripts/create_inspiration_images.py
+++ /dev/null
@@ -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