Add huggingface parms to config file

pull/2292/head
bmaltais 2024-04-14 22:03:36 -04:00
parent 91b679f9ca
commit 5316db3fa7
4 changed files with 13 additions and 3 deletions

View File

@ -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

View File

@ -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"):

View File

@ -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,

View File

@ -5,7 +5,7 @@ from .class_gui_config import KohyaSSGUIConfig
class HuggingFace:
def __init__(
self,
config: KohyaSSGUIConfig = {},
config: KohyaSSGUIConfig,
) -> None:
self.config = config