Text-to-Image
SEED-Story
English
Andyson commited on
Commit
80fcc80
1 Parent(s): 121cb64

add config.json

Browse files
detokenizer/detokenizer_george_adapted/checkpoint-4000/config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "sdxl_adapter",
3
+ "resampler": {
4
+ "dim": 1024,
5
+ "depth": 4,
6
+ "dim_head": 64,
7
+ "heads": 16,
8
+ "num_queries": 64,
9
+ "embedding_dim": 4096,
10
+ "output1_dim": 768,
11
+ "output2_dim": 1280,
12
+ "ff_mult": 4
13
+ },
14
+ "pretrained_model_path": "TencentARC/SEED-Story",
15
+ "subfolder": "detokenizer/detokenizer_george_adapted/checkpoint-4000"
16
+ }
seed_story/george_sft/config.json ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "llama", // Assuming the base model type is LLaMA; replace if different
3
+ "input_resampler": {
4
+ "grid_size": 8,
5
+ "embed_dim": 4096,
6
+ "num_heads": 32,
7
+ "kv_dim": 4096
8
+ },
9
+ "output_resampler": {
10
+ "grid_size": 16,
11
+ "embed_dim": 4096,
12
+ "num_heads": 32,
13
+ "kv_dim": 4096
14
+ },
15
+ "lm_loss_scale": 1.0,
16
+ "rec_loss_scale": 1.0
17
+ }