upload to HF

anima_sdxl_config/config_file.toml

@@ -0,0 +1,67 @@
+[sdxl_arguments]
+cache_text_encoder_outputs = true
+enable_bucket = true
+no_half_vae = true
+cache_latents = true
+cache_latents_to_disk = true
+vae_batch_size = 4
+min_timestep = 0
+max_timestep = 1000
+shuffle_caption = false
+
+[model_arguments]
+pretrained_model_name_or_path = "/content/pretrained_model/sd_xl_base_0.9.safetensors"
+vae = "/content/vae/sdxl_vae.safetensors"
+
+[dataset_arguments]
+debug_dataset = false
+in_json = "/content/fine_tune/meta_clean.json"
+train_data_dir = "/content/fine_tune/train_data"
+dataset_repeats = 1
+keep_tokens = 0
+resolution = "1024,1024"
+caption_dropout_rate = 0
+caption_tag_dropout_rate = 0
+caption_dropout_every_n_epochs = 0
+color_aug = false
+token_warmup_min = 1
+token_warmup_step = 0
+
+[training_arguments]
+output_dir = "/content/fine_tune/output"
+output_name = "anima_sdxl"
+save_precision = "fp16"
+save_every_n_steps = 1000
+train_batch_size = 4
+max_token_length = 225
+mem_eff_attn = false
+xformers = true
+max_train_steps = 10000
+max_data_loader_n_workers = 8
+persistent_data_loader_workers = true
+gradient_checkpointing = true
+gradient_accumulation_steps = 1
+mixed_precision = "no"
+
+[logging_arguments]
+log_with = "tensorboard"
+logging_dir = "/content/fine_tune/logs"
+log_prefix = "anima_sdxl"
+
+[sample_prompt_arguments]
+sample_every_n_steps = 100
+sample_sampler = "euler_a"
+
+[saving_arguments]
+save_model_as = "safetensors"
+
+[optimizer_arguments]
+optimizer_type = "AdaFactor"
+learning_rate = 1e-6
+max_grad_norm = 1.0
+optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False",]
+lr_scheduler = "cosine_with_restarts"
+lr_warmup_steps = 100
+lr_scheduler_num_cycles = 0
+
+[advanced_training_config]

anima_sdxl_config/sample_prompt.toml

@@ -0,0 +1,9 @@
+[prompt]
+negative_prompt = " 3d render, smooth, plastic, blurry, grainy, low-resolution, deep-fried, oversaturated"
+width = 1024
+height = 1024
+scale = 7
+sample_steps = 28
+[[prompt.subset]]
+prompt = "masterpiece, best quality, a woman in a red and white outfit is dancing in the air with a moon in the background and a red and gold scarf"
+