Upload folder using huggingface_hub

README.md

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

config.yaml

@@ -12,7 +12,7 @@ config:
       linear_alpha: 16
     save:
       dtype: float16
-      save_every: 501
+      save_every: 101
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
@@ -27,7 +27,7 @@ config:
       - 1024
     train:
       batch_size: 1
-      steps: 500
+      steps: 100
       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.001
+      lr: 0.0001
       ema_config:
         use_ema: true
         ema_decay: 0.99
@@ -46,7 +46,7 @@ config:
       quantize: true
     sample:
       sampler: flowmatch
-      sample_every: 501
+      sample_every: 101
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86d0e5f265b76a14e1deec0632dd380146db8382fbce73a96fb94fd469f0b300
+oid sha256:2750f665fc925f0d173cea656d75c4056d10ec7d3cde987a04f7e8ae386209ea
 size 171969408