Upload 3 files

config.example.toml

@@ -0,0 +1,41 @@
+
+pretrained_model_name_or_path = "path/to/graupel-v1-nobody-fp32.safetensors"
+
+output_dir = "path/to/emb_graupel"
+
+# hyperparameters
+learning_rate = 6e-4
+lr_scheduler = "cosine_with_restarts"
+lr_scheduler_num_cycles = 20
+lr_warmup_steps = 50
+max_train_epochs = 20
+optimizer_type = "Lion"
+seed = 3047
+
+mixed_precision = "fp16"
+xformers = true
+
+save_every_n_epochs = 2
+
+# textual inversion
+init_word = "horror"
+num_vectors_per_token = 16
+token_string = "badquality"
+
+# v2
+v2 = true
+v_parameterization = true
+
+# logging
+log_prefix = "test-13-"
+log_tracker_name = "badquality-fix"
+log_with = "wandb"
+# logging_dir = ""
+
+# sample
+sample_every_n_epochs = 1
+sample_prompts = "path/to/sample.txt"
+
+# dataset
+dataset_config = "path/to/dataset.toml"
+debug_dataset = false

dataset.example.toml

@@ -0,0 +1,41 @@
+[general]
+caption_extension = '.txt'
+enable_bucket = true
+shuffle_caption = true
+
+# これは DreamBooth 方式のデータセット
+[[datasets]]
+batch_size = 2
+resolution = 640
+
+[[datasets.subsets]]
+image_dir = 'path/to/carpet'
+num_repeats = 4
+
+[[datasets.subsets]]
+image_dir = 'path/to/bad'
+num_repeats = 2
+
+[[datasets.subsets]]
+image_dir = 'path/to/bad anatomy'
+num_repeats = 2
+
+[[datasets.subsets]]
+image_dir = 'path/to/carpet'
+num_repeats = 2
+
+[[datasets.subsets]]
+image_dir = 'path/to/1_base blurry 50'
+num_repeats = 2
+
+[[datasets.subsets]]
+image_dir = 'path/to/2_bad anatomy 50'
+num_repeats = 2
+
+[[datasets.subsets]]
+image_dir = 'path/to/2_signature 20'
+num_repeats = 1
+
+[[datasets.subsets]]
+image_dir = 'path/to/3_base 50'
+num_repeats = 2

sample.txt

@@ -0,0 +1,3 @@
+masterpiece, best quality, best aesthetic, 1girl, solo, upper body --n badquality --h 768 --s 25
+masterpiece, best quality, best aesthetic, 1girl, blue hair, cat ears, white dress shirt, upper body, portrait, looking at viewer, white background --n badquality --h 768 --d 999 --s 25
+badquality