Upload folder using huggingface_hub

.ipynb_checkpoints/config-checkpoint.toml

@@ -0,0 +1,11 @@
+[general]
+enable_bucket = true                        # Whether to use Aspect Ratio Bucketing
+
+[[datasets]]
+resolution = 512                            # Training resolution
+batch_size = 4                              # Batch size
+
+  [[datasets.subsets]]
+  image_dir = '/workspace/sd-scripts/tuan'                     # Specify the folder containing the training images
+  # caption_extension = '.caption'            # Caption file extension; change this if using .txt
+  num_repeats = 10                          # Number of repetitions for training images

.ipynb_checkpoints/script-checkpoint.sh

@@ -0,0 +1,15 @@
+#!usr/bin/bash
+accelerate launch --num_cpu_threads_per_process 1 train_db.py 
+    --pretrained_model_name_or_path=runwayml/stable-diffusion-v1-5
+    --dataset_config=config.toml
+    --output_dir=db-kohya-tuandao-folder
+    --output_name=db-kohya-tuandao
+    --save_model_as=safetensors 
+    --prior_loss_weight=1.0 
+    --max_train_steps=1600 
+    --learning_rate=1e-6 
+    --optimizer_type="AdamW8bit" 
+    --xformers 
+    --mixed_precision="fp16" 
+    --cache_latents 
+    --gradient_checkpointing

config.toml

@@ -0,0 +1,11 @@
+[general]
+enable_bucket = true                        # Whether to use Aspect Ratio Bucketing
+
+[[datasets]]
+resolution = 512                            # Training resolution
+batch_size = 4                              # Batch size
+
+  [[datasets.subsets]]
+  image_dir = '/workspace/sd-scripts/tuan'                     # Specify the folder containing the training images
+  # caption_extension = '.caption'            # Caption file extension; change this if using .txt
+  num_repeats = 10                          # Number of repetitions for training images

db-kohya-tuandao.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c66c51b0b91305d0f8fb2b2094f1bf68bf2c5c7804a84c33b2580954bb5b0a08
+size 4097789094

script.sh

@@ -0,0 +1,15 @@
+#!usr/bin/bash
+accelerate launch --num_cpu_threads_per_process 1 train_db.py \
+    --pretrained_model_name_or_path="runwayml/stable-diffusion-v1-5" \
+    --dataset_config=config.toml \
+    --output_dir=db-kohya-tuandao-folder \
+    --output_name=db-kohya-tuandao \
+    --save_model_as=safetensors \
+    --prior_loss_weight=1.0 \
+    --max_train_steps=1600 \
+    --learning_rate=1e-6 \
+    --optimizer_type="AdamW8bit" \
+    --xformers \
+    --mixed_precision="fp16" \
+    --cache_latents \
+    --gradient_checkpointing