Andrew Powers-Holmes commited on
Commit
bed8171
1 Parent(s): 8067e66

add kl-f8-anime2

Browse files
kl-f8-anime2/config.yaml ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 4.5e-06
3
+ target: ldm.models.autoencoder.AutoencoderKL
4
+ params:
5
+ monitor: val/rec_loss
6
+ embed_dim: 4
7
+ lossconfig:
8
+ target: ldm.modules.losses.LPIPSWithDiscriminator
9
+ params:
10
+ disc_start: 50001
11
+ kl_weight: 1.0e-06
12
+ disc_weight: 0.5
13
+ ddconfig:
14
+ double_z: true
15
+ z_channels: 4
16
+ resolution: 256
17
+ in_channels: 3
18
+ out_ch: 3
19
+ ch: 128
20
+ ch_mult:
21
+ - 1
22
+ - 2
23
+ - 4
24
+ - 4
25
+ num_res_blocks: 2
26
+ attn_resolutions: []
27
+ dropout: 0.0
28
+ data:
29
+ target: main.DataModuleFromConfig
30
+ params:
31
+ batch_size: 4
32
+ wrap: true
33
+ train:
34
+ target: ldm.data.openimages.FullOpenImagesTrain
35
+ params:
36
+ size: 384
37
+ crop_size: 256
38
+ validation:
39
+ target: ldm.data.openimages.FullOpenImagesValidation
40
+ params:
41
+ size: 384
42
+ crop_size: 256
kl-f8-anime2/model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df3c506e51b7ee1d7b5a6a2bb7142d47d488743c96aa778afb0f53a2cdc2d38d
3
+ size 404662241