Hugging Face's logo

Spaces:
shi-labs
/
Prompt-Free-Diffusion
Runtime error

App Files Community
2
Prompt-Free-Diffusion
File size: 1,320 Bytes 
515f781
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
 
seecoder_base:
  symbol: seecoder
  args: {}

seecoder:
  super_cfg: seecoder_base
  type: seecoder
  args:
    imencoder_cfg : MODEL(swin_large)
    imdecoder_cfg : MODEL(seecoder_decoder)
    qtransformer_cfg : MODEL(seecoder_query_transformer)

seecoder_pa:
  super_cfg: seet
  type: seecoder
  args:
    imencoder_cfg : MODEL(swin_large)
    imdecoder_cfg : MODEL(seecoder_decoder)
    qtransformer_cfg : MODEL(seecoder_query_transformer_position_aware)

###########
# decoder #
###########

seecoder_decoder:
  super_cfg: seecoder_base
  type: seecoder_decoder
  args:
    inchannels:
      res3:  384
      res4:  768
      res5: 1536
    trans_input_tags: ['res3', 'res4', 'res5']
    trans_dim: 768
    trans_dropout: 0.1
    trans_nheads: 8
    trans_feedforward_dim: 1024
    trans_num_layers: 6

#####################
# query_transformer #
#####################

seecoder_query_transformer:
  super_cfg: seecoder_base
  type: seecoder_query_transformer
  args:
    in_channels : 768
    hidden_dim: 768
    num_queries: [4, 144]
    nheads: 8
    num_layers: 9
    feedforward_dim: 2048
    pre_norm: False
    num_feature_levels: 3
    enforce_input_project: False
    with_fea2d_pos: false

seecoder_query_transformer_position_aware:
  super_cfg: seecoder_query_transformer
  args:
    with_fea2d_pos: true