BAAI
/

shunxing1234 commited on
Commit
3328559
1 Parent(s): e47bed9

Upload 4 files

Browse files
text_encoder/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "xlm-roberta-large",
3
+ "architectures": [
4
+ "RobertaSeriesModelWithTransformation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "bos_token_id": 0,
8
+ "classifier_dropout": null,
9
+ "eos_token_id": 2,
10
+ "hidden_act": "gelu",
11
+ "hidden_dropout_prob": 0.0,
12
+ "hidden_size": 1024,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 4096,
15
+ "layer_norm_eps": 1e-05,
16
+ "learn_encoder": false,
17
+ "max_position_embeddings": 514,
18
+ "model_type": "xlm-roberta",
19
+ "num_attention_heads": 16,
20
+ "num_hidden_layers": 24,
21
+ "output_past": true,
22
+ "pad_token_id": 1,
23
+ "pooler_fn": "cls",
24
+ "position_embedding_type": "absolute",
25
+ "project_dim": 1024,
26
+ "torch_dtype": "float32",
27
+ "transformers_version": "4.24.0.dev0",
28
+ "type_vocab_size": 1,
29
+ "use_attention_mask": true,
30
+ "use_cache": true,
31
+ "vocab_size": 250002
32
+ }
text_encoder/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eabb14dbd598313315e08d617290fbe33f771ee63e95e8744cd345a75f6cb630
3
+ size 2243903745
unet/config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.14.0",
4
+ "act_fn": "silu",
5
+ "attention_head_dim": [
6
+ 5,
7
+ 10,
8
+ 20,
9
+ 20
10
+ ],
11
+ "block_out_channels": [
12
+ 320,
13
+ 640,
14
+ 1280,
15
+ 1280
16
+ ],
17
+ "center_input_sample": false,
18
+ "class_embed_type": null,
19
+ "conv_in_kernel": 3,
20
+ "conv_out_kernel": 3,
21
+ "cross_attention_dim": 1024,
22
+ "down_block_types": [
23
+ "CrossAttnDownBlock2D",
24
+ "CrossAttnDownBlock2D",
25
+ "CrossAttnDownBlock2D",
26
+ "DownBlock2D"
27
+ ],
28
+ "downsample_padding": 1,
29
+ "dual_cross_attention": false,
30
+ "flip_sin_to_cos": true,
31
+ "freq_shift": 0,
32
+ "in_channels": 4,
33
+ "layers_per_block": 2,
34
+ "mid_block_scale_factor": 1,
35
+ "mid_block_type": "UNetMidBlock2DCrossAttn",
36
+ "norm_eps": 1e-05,
37
+ "norm_num_groups": 32,
38
+ "num_class_embeds": null,
39
+ "only_cross_attention": false,
40
+ "out_channels": 4,
41
+ "projection_class_embeddings_input_dim": null,
42
+ "resnet_time_scale_shift": "default",
43
+ "sample_size": 96,
44
+ "time_cond_proj_dim": null,
45
+ "time_embedding_type": "positional",
46
+ "timestep_post_act": null,
47
+ "up_block_types": [
48
+ "UpBlock2D",
49
+ "CrossAttnUpBlock2D",
50
+ "CrossAttnUpBlock2D",
51
+ "CrossAttnUpBlock2D"
52
+ ],
53
+ "upcast_attention": false,
54
+ "use_linear_projection": true
55
+ }
unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d2c2e2537f67fbca7e8d7f18861ed6aed4c823d76613d4d93211ba1bea6f0c3
3
+ size 3463913445