File size: 1,775 Bytes
004fe4b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
#########
# v1 2d #
#########

openai_unet_2d_v1:
  type: openai_unet_2d_next
  args:
    in_channels: 4
    out_channels: 4
    model_channels: 320
    attention_resolutions: [ 4, 2, 1 ]
    num_res_blocks: [ 2, 2, 2, 2 ]
    channel_mult: [ 1, 2, 4, 4 ]
    num_heads: 8
    context_dim: 768
    use_checkpoint: True
    parts: [global, data, context]

openai_unet_2d_v1_g:
  super_cfg: openai_unet_2d_v1
  args:
    parts: [global]

openai_unet_2d_v1_d:
  super_cfg: openai_unet_2d_v1
  args:
    parts: [data]

openai_unet_2d_v1_c:
  super_cfg: openai_unet_2d_v1
  args:
    parts: [context]

openai_unet_2d_v1_gd:
  super_cfg: openai_unet_2d_v1
  args:
    parts: [global, data]

openai_unet_2d_v1_gc:
  super_cfg: openai_unet_2d_v1
  args:
    parts: [global, context]

openai_unet_2d_v1_dc:
  super_cfg: openai_unet_2d_v1
  args:
    parts: [data, context]

#########
# v1 0d #
#########

openai_unet_0d_v1:
  type: openai_unet_0d_next
  args:
    input_channels: 768
    model_channels: 320
    output_channels: 768
    num_noattn_blocks: [ 2, 2, 2, 2 ]
    channel_mult: [ 1, 2, 4, 4 ]
    second_dim: [ 4, 4, 4, 4 ]
    with_attn: [true, true, true, false]
    num_heads: 8
    context_dim: 768
    use_checkpoint: True
    parts: [global, data, context]

openai_unet_0d_v1_g:
  super_cfg: openai_unet_0d_v1
  args:
    parts: [global]

openai_unet_0d_v1_d:
  super_cfg: openai_unet_0d_v1
  args:
    parts: [data]

openai_unet_0d_v1_c:
  super_cfg: openai_unet_0d_v1
  args:
    parts: [context]

openai_unet_0d_v1_gd:
  super_cfg: openai_unet_0d_v1
  args:
    parts: [global, data]

openai_unet_0d_v1_gc:
  super_cfg: openai_unet_0d_v1
  args:
    parts: [global, context]

openai_unet_0d_v1_dc:
  super_cfg: openai_unet_0d_v1
  args:
    parts: [data, context]