hf upload

anima-diffusion_config/config_file.toml

@@ -0,0 +1,63 @@
+[sdxl_arguments]
+cache_text_encoder_outputs = true
+no_half_vae = false
+min_timestep = 0
+max_timestep = 1000
+shuffle_caption = false
+
+[model_arguments]
+pretrained_model_name_or_path = "/content/pretrained_model/wdxl-aesthetic-0.9.safetensors"
+vae = "/content/vae/sdxl_vae.safetensors"
+
+[dataset_arguments]
+debug_dataset = false
+in_json = "/content/fine_tune/meta_lat.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-diffusion"
+save_precision = "fp16"
+save_every_n_steps = 2500
+train_batch_size = 4
+max_token_length = 225
+mem_eff_attn = false
+xformers = true
+max_train_steps = 2500
+max_data_loader_n_workers = 8
+persistent_data_loader_workers = true
+gradient_checkpointing = true
+gradient_accumulation_steps = 1
+mixed_precision = "fp16"
+
+[logging_arguments]
+log_with = "tensorboard"
+logging_dir = "/content/fine_tune/logs"
+log_prefix = "anima-diffusion"
+
+[sample_prompt_arguments]
+sample_every_n_steps = 250
+sample_sampler = "euler_a"
+
+[saving_arguments]
+save_model_as = "safetensors"
+
+[optimizer_arguments]
+optimizer_type = "AdaFactor"
+learning_rate = 4e-7
+train_text_encoder = false
+max_grad_norm = 0
+optimizer_args = [ "scale_parameter=False", "relative_step=False", "warmup_init=False",]
+lr_scheduler = "constant_with_warmup"
+lr_warmup_steps = 100
+
+[advanced_training_config]

anima-diffusion_config/sample_prompt.toml

@@ -0,0 +1,14 @@
+[prompt]
+width = 1024
+height = 1024
+scale = 7
+sample_steps = 28
+[[prompt.subset]]
+prompt = "anime, 1girl, solo, long hair, breasts, looking at viewer, blush, large breasts, cleavage, sitting, purple eyes, collarbone, swimsuit, outdoors, sky, day, one-piece swimsuit, ocean, beach"
+
+[[prompt.subset]]
+prompt = "anime, 1girl, solo, long hair, looking at viewer, blush, open mouth, skirt, black hair, school uniform, yellow eyes, ass, censored, pussy, socks, cum, mosaic censoring, sweater, blue skirt, anus, cum in pussy, black socks, after sex, yellow sweater"
+
+[[prompt.subset]]
+prompt = "anime, 1girl, solo, long hair, breasts, looking at viewer, large breasts, cleavage, purple eyes, upper body, ponytail, purple hair, horns, formal, suit, single horn, oni horns, green shirt, purple jacket"
+