mirror of https://github.com/bmaltais/kohya_ss
37 lines
1.5 KiB
Python
37 lines
1.5 KiB
Python
import gradio as gr
|
|
|
|
class SDXLParameters:
|
|
def __init__(
|
|
self,
|
|
sdxl_checkbox: gr.Checkbox,
|
|
show_sdxl_cache_text_encoder_outputs: bool = True,
|
|
):
|
|
self.sdxl_checkbox = sdxl_checkbox
|
|
self.show_sdxl_cache_text_encoder_outputs = (
|
|
show_sdxl_cache_text_encoder_outputs
|
|
)
|
|
self.initialize_accordion()
|
|
|
|
def initialize_accordion(self):
|
|
with gr.Accordion(
|
|
visible=False, open=True, label='SDXL Specific Parameters'
|
|
) as self.sdxl_row:
|
|
with gr.Row():
|
|
self.sdxl_cache_text_encoder_outputs = gr.Checkbox(
|
|
label='Cache text encoder outputs',
|
|
info='Cache the outputs of the text encoders. This option is useful to reduce the GPU memory usage. This option cannot be used with options for shuffling or dropping the captions.',
|
|
value=False,
|
|
visible=self.show_sdxl_cache_text_encoder_outputs,
|
|
)
|
|
self.sdxl_no_half_vae = gr.Checkbox(
|
|
label='No half VAE',
|
|
info='Disable the half-precision (mixed-precision) VAE. VAE for SDXL seems to produce NaNs in some cases. This option is useful to avoid the NaNs.',
|
|
value=True,
|
|
)
|
|
|
|
self.sdxl_checkbox.change(
|
|
lambda sdxl_checkbox: gr.Accordion(visible=sdxl_checkbox),
|
|
inputs=[self.sdxl_checkbox],
|
|
outputs=[self.sdxl_row],
|
|
)
|