update

hparams.yml

@@ -0,0 +1,46 @@
+adam_beta1: 0.9
+adam_beta2: 0.99
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.01
+adam_weight_decay_text_encoder: null
+allow_tf32: false
+blur_kernel_size_range: 21@51
+budget: 2
+censor_modes: pixelation@gaussian_blur@black
+center_crop: false
+checkpointing_steps: 1000
+cover_image_column: null
+cover_image_mode: censorship
+dataloader_num_workers: 0
+dataset_name: timm/imagenet-1k-wds
+disable_n_sq: false
+enable_n_crop: false
+enable_n_fq: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+image_log_steps: 50
+jpeg_qf: 90
+learning_rate: 1.0
+logging_dir: logs
+make_cover_image: true
+max_grad_norm: 1.0
+max_train_samples: null
+max_train_steps: 10000
+message_image_column: jpg
+mixed_precision: null
+num_train_epochs: 1
+output_dir: /work/ernestchu0/censorship/outputs
+pixelation_block_size_range: 10@40
+pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
+prodigy_beta3: null
+prodigy_decouple: true
+prodigy_safeguard_warmup: true
+prodigy_use_bias_correction: true
+random_flip: false
+rank: 64
+report_to: tensorboard
+resolution: 520
+resume_from_checkpoint: null
+seed: 12345
+train_batch_size: 2
+train_data_size: null

pytorch_lora_weights.safetensors

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