akon1995 commited on
Commit
db0e0d2
·
verified ·
1 Parent(s): e7fac1c

Upload ControlRadio-SRM pretrained checkpoint

Browse files
experiments/prompt_v6/RadioMapSeer_RadioDiff-Seer-no-carsInput/control_sd21_3ch_1e-05_3_100_sd_tune_seed1230/lightning_logs/version_0/checkpoints/epoch=99-step=337499.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea8fe13df69a57b8714a6af2daab8078e03de81932fce1520ff71c5e9da16fea
3
+ size 13700533604
experiments/prompt_v6/RadioMapSeer_RadioDiff-Seer-no-carsInput/control_sd21_3ch_1e-05_3_100_sd_tune_seed1230/lightning_logs/version_0/hparams.yaml ADDED
@@ -0,0 +1,91 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ control_stage_config:
2
+ target: cldm.cldm.ControlNet
3
+ params:
4
+ use_checkpoint: true
5
+ image_size: 32
6
+ in_channels: 4
7
+ hint_channels: 3
8
+ model_channels: 320
9
+ attention_resolutions:
10
+ - 4
11
+ - 2
12
+ - 1
13
+ num_res_blocks: 2
14
+ channel_mult:
15
+ - 1
16
+ - 2
17
+ - 4
18
+ - 4
19
+ num_head_channels: 64
20
+ use_spatial_transformer: true
21
+ use_linear_in_transformer: true
22
+ transformer_depth: 1
23
+ context_dim: 1024
24
+ legacy: false
25
+ control_key: hint
26
+ only_mid_control: false
27
+ linear_start: 0.00085
28
+ linear_end: 0.012
29
+ num_timesteps_cond: 1
30
+ log_every_t: 200
31
+ timesteps: 1000
32
+ first_stage_key: jpg
33
+ cond_stage_key: txt
34
+ image_size: 64
35
+ channels: 4
36
+ cond_stage_trainable: false
37
+ conditioning_key: crossattn
38
+ monitor: val/loss_simple_ema
39
+ scale_factor: 0.18215
40
+ use_ema: false
41
+ unet_config:
42
+ target: cldm.cldm.ControlledUnetModel
43
+ params:
44
+ use_checkpoint: true
45
+ image_size: 32
46
+ in_channels: 4
47
+ out_channels: 4
48
+ model_channels: 320
49
+ attention_resolutions:
50
+ - 4
51
+ - 2
52
+ - 1
53
+ num_res_blocks: 2
54
+ channel_mult:
55
+ - 1
56
+ - 2
57
+ - 4
58
+ - 4
59
+ num_head_channels: 64
60
+ use_spatial_transformer: true
61
+ use_linear_in_transformer: true
62
+ transformer_depth: 1
63
+ context_dim: 1024
64
+ legacy: false
65
+ first_stage_config:
66
+ target: ldm.models.autoencoder.AutoencoderKL
67
+ params:
68
+ embed_dim: 4
69
+ monitor: val/rec_loss
70
+ ddconfig:
71
+ double_z: true
72
+ z_channels: 4
73
+ resolution: 256
74
+ in_channels: 3
75
+ out_ch: 3
76
+ ch: 128
77
+ ch_mult:
78
+ - 1
79
+ - 2
80
+ - 4
81
+ - 4
82
+ num_res_blocks: 2
83
+ attn_resolutions: []
84
+ dropout: 0.0
85
+ lossconfig:
86
+ target: torch.nn.Identity
87
+ cond_stage_config:
88
+ target: ldm.modules.encoders.modules.FrozenOpenCLIPEmbedder
89
+ params:
90
+ freeze: true
91
+ layer: penultimate