ernestchu commited on
Commit
3319577
1 Parent(s): 7692e86

add ip2p w

Browse files
ip2p_hparams.yml ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: 5
9
+ censor_modes: pixelation@gaussian_blur@black
10
+ center_crop: false
11
+ checkpointing_steps: 500
12
+ dataloader_num_workers: 0
13
+ dataset_name: timbrooks/instructpix2pix-clip-filtered
14
+ gradient_accumulation_steps: 4
15
+ gradient_checkpointing: false
16
+ image_column: original_image
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: 10000
23
+ mixed_precision: null
24
+ num_train_epochs: 5
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: 520
36
+ resume_from_checkpoint: null
37
+ seed: null
38
+ target_image_column: edited_image
39
+ train_batch_size: 2
ip2p_pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:921d0640663426e275ae711dce5ad94f50707a2bdaceabbdf7c4271c438d5d04
3
+ size 54215768