leonelhs commited on
Commit
b97d8c4
·
1 Parent(s): cad7340

init inpainting

Browse files
inpainting_big/config.yaml ADDED
@@ -0,0 +1,67 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ base_learning_rate: 1.0e-06
3
+ target: ldm.models.diffusion.ddpm.LatentDiffusion
4
+ params:
5
+ linear_start: 0.0015
6
+ linear_end: 0.0205
7
+ log_every_t: 100
8
+ timesteps: 1000
9
+ loss_type: l1
10
+ first_stage_key: image
11
+ cond_stage_key: masked_image
12
+ image_size: 64
13
+ channels: 3
14
+ concat_mode: true
15
+ monitor: val/loss
16
+ scheduler_config:
17
+ target: ldm.lr_scheduler.LambdaWarmUpCosineScheduler
18
+ params:
19
+ verbosity_interval: 0
20
+ warm_up_steps: 1000
21
+ max_decay_steps: 50000
22
+ lr_start: 0.001
23
+ lr_max: 0.1
24
+ lr_min: 0.0001
25
+ unet_config:
26
+ target: ldm.modules.diffusionmodules.openaimodel.UNetModel
27
+ params:
28
+ image_size: 64
29
+ in_channels: 7
30
+ out_channels: 3
31
+ model_channels: 256
32
+ attention_resolutions:
33
+ - 8
34
+ - 4
35
+ - 2
36
+ num_res_blocks: 2
37
+ channel_mult:
38
+ - 1
39
+ - 2
40
+ - 3
41
+ - 4
42
+ num_heads: 8
43
+ resblock_updown: true
44
+ first_stage_config:
45
+ target: ldm.models.autoencoder.VQModelInterface
46
+ params:
47
+ embed_dim: 3
48
+ n_embed: 8192
49
+ monitor: val/rec_loss
50
+ ddconfig:
51
+ attn_type: none
52
+ double_z: false
53
+ z_channels: 3
54
+ resolution: 256
55
+ in_channels: 3
56
+ out_ch: 3
57
+ ch: 128
58
+ ch_mult:
59
+ - 1
60
+ - 2
61
+ - 4
62
+ num_res_blocks: 2
63
+ attn_resolutions: []
64
+ dropout: 0.0
65
+ lossconfig:
66
+ target: ldm.modules.losses.contperceptual.DummyLoss
67
+ cond_stage_config: __is_first_stage__
inpainting_big/last.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ec79e83b90594096f0bff19efb21e1da5ff62ffbd0291a6b3057fce3213ba3a
3
+ size 3311259854