Merge branch 'dev' into test-fp8

        self.lin_module = None

        self.org_module: list[torch.Module] = [self.sd_module]

        self.scale = 1.0

        # kohya-ss

        if "oft_blocks" in weights.w.keys():

            self.is_kohya = True

            self.constraint = None

            self.block_size, self.num_blocks = factorization(self.out_dim, self.dim)

    def calc_updown_kb(self, orig_weight, multiplier):

    def calc_updown(self, orig_weight):

        oft_blocks = self.oft_blocks.to(orig_weight.device, dtype=orig_weight.dtype)

        oft_blocks = oft_blocks - oft_blocks.transpose(1, 2) # ensure skew-symmetric orthogonal matrix

        eye = torch.eye(self.block_size, device=self.oft_blocks.device)

        if self.is_kohya:

            block_Q = oft_blocks - oft_blocks.transpose(1, 2) # ensure skew-symmetric orthogonal matrix

            norm_Q = torch.norm(block_Q.flatten())

            new_norm_Q = torch.clamp(norm_Q, max=self.constraint)

            block_Q = block_Q * ((new_norm_Q + 1e-8) / (norm_Q + 1e-8))

            oft_blocks = torch.matmul(eye + block_Q, (eye - block_Q).float().inverse())

        R = oft_blocks.to(orig_weight.device, dtype=orig_weight.dtype)

        R = R * multiplier + torch.eye(self.block_size, device=orig_weight.device)

        # This errors out for MultiheadAttention, might need to be handled up-stream

        merged_weight = rearrange(orig_weight, '(k n) ... -> k n ...', k=self.num_blocks, n=self.block_size)

        updown = merged_weight.to(orig_weight.device, dtype=orig_weight.dtype) - orig_weight

        output_shape = orig_weight.shape

        return self.finalize_updown(updown, orig_weight, output_shape)

    def calc_updown(self, orig_weight):

        # if alpha is a very small number as in coft, calc_scale() will return a almost zero number so we ignore it

        multiplier = self.multiplier()

        return self.calc_updown_kb(orig_weight, multiplier)

    # override to remove the multiplier/scale factor; it's already multiplied in get_weight

    def finalize_updown(self, updown, orig_weight, output_shape, ex_bias=None):

        if self.bias is not None:

            updown = updown.reshape(self.bias.shape)

            updown += self.bias.to(orig_weight.device, dtype=orig_weight.dtype)

            updown = updown.reshape(output_shape)

        if len(output_shape) == 4:

            updown = updown.reshape(output_shape)

        if orig_weight.size().numel() == updown.size().numel():

            updown = updown.reshape(orig_weight.shape)

        if ex_bias is not None:

            ex_bias = ex_bias * self.multiplier()

        return updown, ex_bias

    bundle_embeddings = {}

    for key_network, weight in sd.items():

        key_network_without_network_parts, network_part = key_network.split(".", 1)

        key_network_without_network_parts, _, network_part = key_network.partition(".")

        if key_network_without_network_parts == "bundle_emb":

            emb_name, vec_name = network_part.split(".", 1)

            emb_dict = bundle_embeddings.get(emb_name, {})

        self.setting_names = []

        self.infotext_fields = []

        extra_options = shared.opts.extra_options_img2img if is_img2img else shared.opts.extra_options_txt2img

        elem_id_tabname = "extra_options_" + ("img2img" if is_img2img else "txt2img")

        mapping = {k: v for v, k in generation_parameters_copypaste.infotext_to_setting_name_mapping}

        with gr.Blocks() as interface:

            with gr.Accordion("Options", open=False) if shared.opts.extra_options_accordion and extra_options else gr.Group():

            with gr.Accordion("Options", open=False, elem_id=elem_id_tabname) if shared.opts.extra_options_accordion and extra_options else gr.Group(elem_id=elem_id_tabname):

                row_count = math.ceil(len(extra_options) / shared.opts.extra_options_cols)

"""),

    "extra_options_txt2img": shared.OptionInfo([], "Settings for txt2img", ui_components.DropdownMulti, lambda: {"choices": list(shared.opts.data_labels.keys())}).js("info", "settingsHintsShowQuicksettings").info("setting entries that also appear in txt2img interfaces").needs_reload_ui(),

    "extra_options_img2img": shared.OptionInfo([], "Settings for img2img", ui_components.DropdownMulti, lambda: {"choices": list(shared.opts.data_labels.keys())}).js("info", "settingsHintsShowQuicksettings").info("setting entries that also appear in img2img interfaces").needs_reload_ui(),

    "extra_options_cols": shared.OptionInfo(1, "Number of columns for added settings", gr.Number, {"precision": 0}).needs_reload_ui(),

    "extra_options_cols": shared.OptionInfo(1, "Number of columns for added settings", gr.Slider, {"step": 1, "minimum": 1, "maximum": 20}).info("displayed amount will depend on the actual browser window width").needs_reload_ui(),

    "extra_options_accordion": shared.OptionInfo(False, "Place added settings into an accordion").needs_reload_ui()

}))

    if (modalImage && modalImage.offsetParent) {

        let currentButton = selected_gallery_button();

        let preview = gradioApp().querySelectorAll('.livePreview > img');

        if (preview.length > 0) {

        if (opts.js_live_preview_in_modal_lightbox && preview.length > 0) {

            // show preview image if available

            modalImage.src = preview[preview.length - 1].src;

        } else if (currentButton?.children?.length > 0 && modalImage.src != currentButton.children[0].src) {

}

/**

 * Configure the width and height elements on `tabname` to accept

 * pasting of resolutions in the form of "width x height".

 */

function setupResolutionPasting(tabname) {

    var width = gradioApp().querySelector(`#${tabname}_width input[type=number]`);

    var height = gradioApp().querySelector(`#${tabname}_height input[type=number]`);

    for (const el of [width, height]) {

        el.addEventListener('paste', function(event) {

            var pasteData = event.clipboardData.getData('text/plain');

            var parsed = pasteData.match(/^\s*(\d+)\D+(\d+)\s*$/);

            if (parsed) {

                width.value = parsed[1];

                height.value = parsed[2];

                updateInput(width);

                updateInput(height);

                event.preventDefault();

            }

        });

    }

}

onUiLoaded(function() {

    showRestoreProgressButton('txt2img', localGet("txt2img_task_id"));

    showRestoreProgressButton('img2img', localGet("img2img_task_id"));

    setupResolutionPasting('txt2img');

    setupResolutionPasting('img2img');

});