ernestchu commited on
Commit
58f947a
·
1 Parent(s): e0ae1b6
Files changed (2) hide show
  1. hparams.yml +38 -0
  2. pytorch_lora_weights.safetensors +3 -0
hparams.yml ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.99
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ adam_weight_decay_text_encoder: null
6
+ allow_tf32: false
7
+ blur_kernel_size_range: 21@51
8
+ budget: 1
9
+ censor_modes: pixelation@gaussian_blur@black
10
+ center_crop: false
11
+ checkpointing_steps: 500
12
+ dataloader_num_workers: 0
13
+ dataset_name: timm/imagenet-1k-wds
14
+ gradient_accumulation_steps: 4
15
+ gradient_checkpointing: false
16
+ image_column: jpg
17
+ image_log_steps: 50
18
+ learning_rate: 1.0
19
+ logging_dir: logs
20
+ max_grad_norm: 1.0
21
+ max_train_samples: null
22
+ max_train_steps: 11439
23
+ mixed_precision: null
24
+ num_train_epochs: 1
25
+ output_dir: outputs
26
+ pixelation_block_size_range: 10@40
27
+ pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
28
+ prodigy_beta3: null
29
+ prodigy_decouple: true
30
+ prodigy_safeguard_warmup: true
31
+ prodigy_use_bias_correction: true
32
+ random_flip: false
33
+ rank: 64
34
+ report_to: tensorboard
35
+ resolution: 512
36
+ resume_from_checkpoint: null
37
+ seed: null
38
+ train_batch_size: 4
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a5c209998d101667ee5dfb289b43c7416cd00dee4cae6ba80ffe1c38abc917ec
3
+ size 54215768