jwheo commited on
Commit
461f888
·
verified ·
1 Parent(s): b12283c

Publish masked detail v2 candidate and benchmark

Browse files
configs/hf/detail_branch_v2_masked_photo130k_lsdir.yaml ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project:
2
+ name: detail_branch_v2_masked_photo130k_lsdir_hf
3
+
4
+ autoencoder:
5
+ config: configs/autoencoder_photo10k.yaml
6
+ checkpoint: checkpoints/stage1_autoencoder_best_eval_recon.pt
7
+
8
+ condition_encoder:
9
+ config: configs/latent_pretrain_photo130k_lsdir_dual_multiscale_long.yaml
10
+ checkpoint: checkpoints/stage2_photo130k_lsdir_dual_multiscale_best98000.pt
11
+ trainable: false
12
+
13
+ detail_mask:
14
+ config: configs/hf/detail_mask_predictor_v1.yaml
15
+ checkpoint: checkpoints/detail_mask_predictor_v1_best3250.pt
16
+ trainable: false
17
+ floor: 0.05
18
+
19
+ data:
20
+ hr_size: 512
21
+ scale: 4
22
+ degradation_preset: photo_detail_mix
23
+ domains:
24
+ photo: 0
25
+ anime: 1
26
+
27
+ model:
28
+ image_channels: 3
29
+ latent_channels: 16
30
+ hidden_channels: 96
31
+ num_blocks: 18
32
+ norm_groups: 32
33
+ num_domains: 2
34
+ residual_scale: 0.24
35
+ gate_bias: -1.5
36
+ highpass_kernel: 15
37
+ use_condition_latent: true
38
+
39
+ train:
40
+ dtype: bf16
41
+
42
+ inference:
43
+ checkpoint: checkpoints/detail_branch_v2_masked_photo130k_lsdir_best38000.pt