Unverified Commit c664b231 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #3162 from discus0434/master 

Added what I forgot to commit in already merged PR: #3086
parents f510a227 365d4b16

modules/hypernetworks/ui.py

+4 −0
Original line number Diff line number Diff line 
import html

import os

import re



import gradio as gr
@@ -13,6 +14,9 @@ def create_hypernetwork(name, enable_sizes, layer_structure=None, add_layer_norm 
    fn = os.path.join(shared.cmd_opts.hypernetwork_dir, f"{name}.pt")

    assert not os.path.exists(fn), f"file {fn} already exists"



    if type(layer_structure) == str:

        layer_structure = tuple(map(int, re.sub(r'\D', '', layer_structure)))



    hypernet = modules.hypernetworks.hypernetwork.Hypernetwork(

        name=name,

        enable_sizes=[int(x) for x in enable_sizes],

modules/ui.py

+1 −1
Original line number Diff line number Diff line
@@ -1217,7 +1217,7 @@ def create_ui(wrap_gradio_gpu_call): 
                with gr.Tab(label="Create hypernetwork"):

                    new_hypernetwork_name = gr.Textbox(label="Name")

                    new_hypernetwork_sizes = gr.CheckboxGroup(label="Modules", value=["768", "320", "640", "1280"], choices=["768", "320", "640", "1280"])

                    new_hypernetwork_layer_structure = gr.Dropdown(label="Hypernetwork layer structure", choices=[(1, 2, 1), (1, 2, 2, 1), (1, 2, 4, 2, 1)])

                    new_hypernetwork_layer_structure = gr.Textbox("1, 2, 1", label="Enter hypernetwork layer structure", placeholder="1st and last digit must be 1. ex:'1, 2, 1'")

                    new_hypernetwork_add_layer_norm = gr.Checkbox(label="Add layer normalization")



                    with gr.Row():

CRA Git | Maintained and supported by SUSTech CRA and CCSE