AzhureRaven commited on
Commit
9e553d9
·
verified ·
1 Parent(s): a13f07d

Upload rico_diffusion_v1.json

Browse files
Files changed (1) hide show
  1. ed2-config/rico_diffusion_v1.json +37 -0
ed2-config/rico_diffusion_v1.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attn_type": "sdp",
3
+ "batch_size": 7,
4
+ "ckpt_every_n_minutes": null,
5
+ "clip_grad_norm": null,
6
+ "clip_skip": 2,
7
+ "cond_dropout": 0.10,
8
+ "data_root": "/home/azhure/Tesis/save_data",
9
+ "disable_amp": false,
10
+ "disable_textenc_training": false,
11
+ "flip_p": 0.0,
12
+ "gpuid": 0,
13
+ "gradient_checkpointing": true,
14
+ "grad_accum": 1,
15
+ "logdir": "logs",
16
+ "log_step": 25,
17
+ "loss_type": "huber",
18
+ "max_epochs": 5,
19
+ "optimizer_config": "optimizer_rico.json",
20
+ "project_name": "rico_diffusion_v1",
21
+ "resolution": 512,
22
+ "resume_ckpt": "stable-diffusion-v1-5/stable-diffusion-v1-5",
23
+ "sample_prompts": "sample_prompts_rico.json",
24
+ "sample_steps": 15000,
25
+ "save_ckpt_dir": "model_output",
26
+ "save_ckpts_from_n_epochs": 0,
27
+ "save_every_n_epochs": 5,
28
+ "save_optimizer": false,
29
+ "scale_lr": false,
30
+ "seed": 555,
31
+ "keep_tags": 3,
32
+ "shuffle_tags": false,
33
+ "validation_config": "validation_rico.json",
34
+ "zero_frequency_noise_ratio": 0.02,
35
+ "min_snr_gamma": 5.0,
36
+ "enable_zero_terminal_snr": true
37
+ }