Upload folder using huggingface_hub

README.md

@@ -21,7 +21,7 @@ instance_prompt: ztj
 
 # Ztj 21
 
-<!-- <Gallery /> -->
+<Gallery />
 
 Trained on Replicate using:
 

config.yaml

@@ -12,7 +12,7 @@ config:
       linear_alpha: 16
     save:
       dtype: float16
-      save_every: 501
+      save_every: 201
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
@@ -27,7 +27,7 @@ config:
       - 1024
     train:
       batch_size: 1
-      steps: 500
+      steps: 200
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
@@ -35,7 +35,7 @@ config:
       gradient_checkpointing: true
       noise_scheduler: flowmatch
       optimizer: adamw8bit
-      lr: 0.0008
+      lr: 0.0004
       ema_config:
         use_ema: true
         ema_decay: 0.99
@@ -46,7 +46,7 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 501
+      sample_every: 201
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:919f49909e3ef0cb1f1af1fa7119a24d1cc3da4ac995f7efe433128aaca46bb9
+oid sha256:930d7ef5f81795861bdf218be3d6618d88afdd6c262f2c133d3b381860ac7c83
 size 171969408