mirror of https://github.com/bmaltais/kohya_ss
42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
import gradio as gr
|
|
from library.merge_lora_gui import gradio_merge_lora_tab
|
|
from library.svd_merge_lora_gui import gradio_svd_merge_lora_tab
|
|
from library.verify_lora_gui import gradio_verify_lora_tab
|
|
from library.resize_lora_gui import gradio_resize_lora_tab
|
|
from library.extract_lora_gui import gradio_extract_lora_tab
|
|
from library.extract_lycoris_locon_gui import gradio_extract_lycoris_locon_tab
|
|
from library.extract_lora_from_dylora_gui import gradio_extract_dylora_tab
|
|
from library.merge_lycoris_gui import gradio_merge_lycoris_tab
|
|
|
|
# Deprecated code
|
|
from library.dataset_balancing_gui import gradio_dataset_balancing_tab
|
|
from library.dreambooth_folder_creation_gui import (
|
|
gradio_dreambooth_folder_creation_tab,
|
|
)
|
|
|
|
class LoRATools:
|
|
def __init__(self, folders = "", headless:bool = False):
|
|
self.headless = headless
|
|
self.folders = folders
|
|
|
|
gr.Markdown(
|
|
'This section provide LoRA tools to help setup your dataset...'
|
|
)
|
|
gradio_extract_dylora_tab(headless=headless)
|
|
gradio_extract_lora_tab(headless=headless)
|
|
gradio_extract_lycoris_locon_tab(headless=headless)
|
|
gradio_merge_lora_tab(headless=headless)
|
|
gradio_merge_lycoris_tab(headless=headless)
|
|
gradio_svd_merge_lora_tab(headless=headless)
|
|
gradio_resize_lora_tab(headless=headless)
|
|
gradio_verify_lora_tab(headless=headless)
|
|
if folders:
|
|
with gr.Tab('Deprecated'):
|
|
gradio_dreambooth_folder_creation_tab(
|
|
train_data_dir_input=folders.train_data_dir,
|
|
reg_data_dir_input=folders.reg_data_dir,
|
|
output_dir_input=folders.output_dir,
|
|
logging_dir_input=folders.logging_dir,
|
|
headless=headless,
|
|
)
|
|
gradio_dataset_balancing_tab(headless=headless) |