File size: 1,033 Bytes
8741abe |
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 |
wandb:
entity: null
# run_id: askkz9i2
resume: 'auto'
experiment:
project: "demo"
name: "show-o-demo"
output_dir: "show-o-demo"
model:
vq_model:
type: "magvitv2"
vq_model_name: "showlab/magvitv2"
showo:
pretrained_model_path: "showlab/show-o"
w_clip_vit: False
vocab_size: 58498
llm_vocab_size: 50295
llm_model_path: 'microsoft/phi-1_5'
codebook_size: 8192
num_vq_tokens: 256
gradient_checkpointing: True
enable_xformers_memory_efficient_attention: True
dataset:
gen_type: "t2i"
und_type: "large_cap"
params:
batch_size: ${training.batch_size}
shuffle_buffer_size: 1000
num_workers: 32
resolution: 256
pin_memory: True
persistent_workers: True
preprocessing:
max_seq_length: 128
resolution: 256
center_crop: False
random_flip: False
training:
gradient_accumulation_steps: 1
cond_dropout_prob: 0.1
batch_size: 20
|