mirror of https://github.com/bmaltais/kohya_ss
Add huggingface parms to config file
parent
91b679f9ca
commit
5316db3fa7
|
|
@ -126,6 +126,16 @@ reg_images_repeat = 1 # Regularisation images
|
|||
util_regularization_images_repeat_input = 1 # Regularisation images repeat input
|
||||
util_training_images_repeat_input = 40 # Training images repeat input
|
||||
|
||||
[huggingface]
|
||||
async_upload = false # Async upload
|
||||
huggingface_path_in_repo = "" # Huggingface path in repo
|
||||
huggingface_repo_id = "" # Huggingface repo id
|
||||
huggingface_repo_type = "" # Huggingface repo type
|
||||
huggingface_repo_visibility = "" # Huggingface repo visibility
|
||||
huggingface_token = "" # Huggingface token
|
||||
resume_from_huggingface = "" # Resume from huggingface (ex: {repo_id}/{path_in_repo}:{revision}:{repo_type})
|
||||
save_state_to_huggingface = false # Save state to huggingface
|
||||
|
||||
[samples]
|
||||
sample_every_n_steps = 0 # Sample every n steps
|
||||
sample_every_n_epochs = 0 # Sample every n epochs
|
||||
|
|
|
|||
|
|
@ -40,7 +40,7 @@ def UI(**kwargs):
|
|||
css=css, title=f"Kohya_ss GUI {release}", theme=gr.themes.Default()
|
||||
)
|
||||
|
||||
config = KohyaSSGUIConfig(config_file_path=kwargs.get("config_file_path"))
|
||||
config = KohyaSSGUIConfig(config_file_path=kwargs.get("config"))
|
||||
|
||||
with interface:
|
||||
with gr.Tab("Dreambooth"):
|
||||
|
|
|
|||
|
|
@ -302,7 +302,7 @@ class BasicTraining:
|
|||
)
|
||||
# Initialize the stop text encoder training slider
|
||||
self.stop_text_encoder_training = gr.Slider(
|
||||
minimum=0,
|
||||
minimum=-1,
|
||||
maximum=100,
|
||||
value=self.config.get("basic.stop_text_encoder_training", 0),
|
||||
step=1,
|
||||
|
|
|
|||
|
|
@ -5,7 +5,7 @@ from .class_gui_config import KohyaSSGUIConfig
|
|||
class HuggingFace:
|
||||
def __init__(
|
||||
self,
|
||||
config: KohyaSSGUIConfig = {},
|
||||
config: KohyaSSGUIConfig,
|
||||
) -> None:
|
||||
self.config = config
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue