Kalaphant commited on
Commit
cf04f6f
1 Parent(s): b245343

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +45 -0
config.json CHANGED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "text-to-video",
3
+ "model_name": "KalaVids",
4
+ "num_layers": 12,
5
+ "hidden_size": 768,
6
+ "vocab_size": 50257,
7
+ "video_resolution": "1080p",
8
+ "frame_rate": 30,
9
+ "num_frames": 300,
10
+ "num_attention_heads": 12,
11
+ "intermediate_size": 3072,
12
+ "hidden_act": "gelu",
13
+ "initializer_range": 0.02,
14
+ "layer_norm_eps": 1e-12,
15
+ "dropout": 0.1,
16
+ "attention_dropout": 0.1,
17
+ "num_labels": 2, // For classification tasks
18
+ "use_cache": true,
19
+ "bos_token_id": 50256, // Typically the same as vocab_size
20
+ "eos_token_id": 50256, // Typically the same as vocab_size
21
+ "pad_token_id": 0,
22
+ "unk_token_id": 50257, // Typically vocab_size + 1
23
+ "special_tokens_map": {
24
+ "bos_token": "<bos_token>",
25
+ "eos_token": "<eos_token>",
26
+ "pad_token": "<pad_token>",
27
+ "unk_token": "<unk_token>"
28
+ },
29
+ "tokenizer_class": "BertTokenizer",
30
+ "additional_special_tokens": [
31
+ "<special_token_1>",
32
+ "<special_token_2>"
33
+ // Add more special tokens if necessary
34
+ ],
35
+ "task_specific_params": {
36
+ "text-to-video": {
37
+ // Task-specific parameters for text-to-video generation
38
+ "video_resolution": "1080p", // Resolution of the generated video
39
+ "frame_rate": 30, // Frame rate of the generated video
40
+ "num_frames": 300 // Number of frames in the generated video
41
+ // Add more task-specific parameters if necessary
42
+ }
43
+ },
44
+ "device_map": "auto" // Automatically determine the device map for layers
45
+ }