Alex Shonenkov commited on
Commit
e953521
1 Parent(s): b10f8ef

add checkpoints

Browse files
.gitattributes CHANGED
@@ -25,3 +25,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
25
  *.zip filter=lfs diff=lfs merge=lfs -text
26
  *.zstandard filter=lfs diff=lfs merge=lfs -text
27
  *tfevents* filter=lfs diff=lfs merge=lfs -text
28
+ vqgan.gumbelf8-sber.model.ckpt filter=lfs diff=lfs merge=lfs -text
vqgan.gumbelf8-sber.config.yml ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 4.5e-06
3
+ target: taming.models.vqgan.GumbelVQ
4
+ params:
5
+ kl_weight: 1.0e-08
6
+ embed_dim: 256
7
+ n_embed: 8192
8
+ monitor: val/rec_loss
9
+ temperature_scheduler_config:
10
+ target: taming.lr_scheduler.LambdaWarmUpCosineScheduler
11
+ params:
12
+ warm_up_steps: 0
13
+ max_decay_steps: 1000001
14
+ lr_start: 0.9
15
+ lr_max: 0.9
16
+ lr_min: 1.0e-06
17
+ ddconfig:
18
+ double_z: false
19
+ z_channels: 256
20
+ resolution: 256
21
+ in_channels: 3
22
+ out_ch: 3
23
+ ch: 128
24
+ ch_mult:
25
+ - 1
26
+ - 1
27
+ - 2
28
+ - 4
29
+ num_res_blocks: 2
30
+ attn_resolutions:
31
+ - 32
32
+ dropout: 0.0
33
+ lossconfig:
34
+ target: taming.modules.losses.vqperceptual.DummyLoss
vqgan.gumbelf8-sber.model.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7be00096299fe50bbcc1a2ed01fe28561464de5020c7c6b5b4a728d3b515c3fb
3
+ size 346054435