Commit 78f59a4e authored by AUTOMATIC's avatar AUTOMATIC
Browse files

enable compact view for train tab

prevent  previews from ruining hypernetwork training
parent 216f67ec
Loading
Loading
Loading
Loading
+2 −0
Original line number Original line Diff line number Diff line
@@ -715,6 +715,8 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi
                        do_not_save_samples=True,
                        do_not_save_samples=True,
                    )
                    )


                    p.disable_extra_networks = True

                    if preview_from_txt2img:
                    if preview_from_txt2img:
                        p.prompt = preview_prompt
                        p.prompt = preview_prompt
                        p.negative_prompt = preview_negative_prompt
                        p.negative_prompt = preview_negative_prompt
+6 −2
Original line number Original line Diff line number Diff line
@@ -140,6 +140,7 @@ class StableDiffusionProcessing:
        self.override_settings = {k: v for k, v in (override_settings or {}).items() if k not in shared.restricted_opts}
        self.override_settings = {k: v for k, v in (override_settings or {}).items() if k not in shared.restricted_opts}
        self.override_settings_restore_afterwards = override_settings_restore_afterwards
        self.override_settings_restore_afterwards = override_settings_restore_afterwards
        self.is_using_inpainting_conditioning = False
        self.is_using_inpainting_conditioning = False
        self.disable_extra_networks = False


        if not seed_enable_extras:
        if not seed_enable_extras:
            self.subseed = -1
            self.subseed = -1
@@ -567,6 +568,7 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
        with devices.autocast():
        with devices.autocast():
            p.init(p.all_prompts, p.all_seeds, p.all_subseeds)
            p.init(p.all_prompts, p.all_seeds, p.all_subseeds)


            if not p.disable_extra_networks:
                extra_networks.activate(p, extra_network_data)
                extra_networks.activate(p, extra_network_data)


        with open(os.path.join(shared.script_path, "params.txt"), "w", encoding="utf8") as file:
        with open(os.path.join(shared.script_path, "params.txt"), "w", encoding="utf8") as file:
@@ -684,7 +686,9 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
            if opts.grid_save:
            if opts.grid_save:
                images.save_image(grid, p.outpath_grids, "grid", p.all_seeds[0], p.all_prompts[0], opts.grid_format, info=infotext(), short_filename=not opts.grid_extended_filename, p=p, grid=True)
                images.save_image(grid, p.outpath_grids, "grid", p.all_seeds[0], p.all_prompts[0], opts.grid_format, info=infotext(), short_filename=not opts.grid_extended_filename, p=p, grid=True)


    if not p.disable_extra_networks:
        extra_networks.deactivate(p, extra_network_data)
        extra_networks.deactivate(p, extra_network_data)

    devices.torch_gc()
    devices.torch_gc()


    res = Processed(p, output_images, p.all_seeds[0], infotext(), comments="".join(["\n\n" + x for x in comments]), subseed=p.all_subseeds[0], index_of_first_image=index_of_first_image, infotexts=infotexts)
    res = Processed(p, output_images, p.all_seeds[0], infotext(), comments="".join(["\n\n" + x for x in comments]), subseed=p.all_subseeds[0], index_of_first_image=index_of_first_image, infotexts=infotexts)
+1 −1
Original line number Original line Diff line number Diff line
@@ -1259,7 +1259,7 @@ def create_ui():
        with gr.Row().style(equal_height=False):
        with gr.Row().style(equal_height=False):
            gr.HTML(value="<p style='margin-bottom: 0.7em'>See <b><a href=\"https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Textual-Inversion\">wiki</a></b> for detailed explanation.</p>")
            gr.HTML(value="<p style='margin-bottom: 0.7em'>See <b><a href=\"https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Textual-Inversion\">wiki</a></b> for detailed explanation.</p>")


        with gr.Row().style(equal_height=False):
        with gr.Row(variant="compact").style(equal_height=False):
            with gr.Tabs(elem_id="train_tabs"):
            with gr.Tabs(elem_id="train_tabs"):


                with gr.Tab(label="Create embedding"):
                with gr.Tab(label="Create embedding"):