add cropping capability

hparams.yml

@@ -11,7 +11,7 @@ center_crop: false
 checkpointing_steps: 500
 dataloader_num_workers: 0
 dataset_name: timm/imagenet-1k-wds
-gradient_accumulation_steps: 4
+gradient_accumulation_steps: 8
 gradient_checkpointing: false
 image_column: jpg
 image_log_steps: 50
@@ -19,9 +19,9 @@ learning_rate: 1.0
 logging_dir: logs
 max_grad_norm: 1.0
 max_train_samples: null
-max_train_steps: 11439
+max_train_steps: 10000
 mixed_precision: null
-num_train_epochs: 1
+num_train_epochs: 2
 output_dir: outputs
 pixelation_block_size_range: 10@40
 pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
@@ -32,7 +32,7 @@ prodigy_use_bias_correction: true
 random_flip: false
 rank: 64
 report_to: tensorboard
-resolution: 512
+resolution: 520
 resume_from_checkpoint: null
 seed: null
-train_batch_size: 4
+train_batch_size: 2

pytorch_lora_weights.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a5c209998d101667ee5dfb289b43c7416cd00dee4cae6ba80ffe1c38abc917ec
+oid sha256:5aad1e0adfc8ec922fb4089b8e3d798e3e1a5cc187ba2c6bb6f3bac617d4f22f
 size 54215768