ymhao commited on
Commit
eb12a54
·
verified ·
1 Parent(s): de0ad1c

Upload config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.yaml +2 -2
config.yaml CHANGED
@@ -1,7 +1,7 @@
1
  mixed_precision: bf16
2
 
3
  stage1:
4
- _target_: hyper_diff_modulator.hyper_modulator_mask.TransformerV2
5
  context_dim: 4096
6
  d_model: 768
7
  n_heads: 12
@@ -19,7 +19,7 @@ stage1:
19
  out_act: softplus
20
 
21
  stage2:
22
- _target_: hyper_diff_modulator.hyper_modulator_transformer.TransformerV2_stage2
23
  context_dim: 4096
24
  d_model: 768
25
  n_heads: 12
 
1
  mixed_precision: bf16
2
 
3
  stage1:
4
+ _target_: lofa.hyper_modulator_mask.TransformerV2
5
  context_dim: 4096
6
  d_model: 768
7
  n_heads: 12
 
19
  out_act: softplus
20
 
21
  stage2:
22
+ _target_: lofa.hyper_modulator_transformer.TransformerV2_stage2
23
  context_dim: 4096
24
  d_model: 768
25
  n_heads: 12