4m-b_mod7_500b--spec_text2im_100b.yaml

# Config for DDP

# Adaptation of 4M-B model for text2im (biased towards captions as input)
run_name: auto

# Input
finetune: '/path/to/trained/4m-b/checkpoint' # Change me

# Input & Output
min_input_tokens: 64
num_input_tokens: 256
min_target_tokens: 256
num_target_tokens: 256
loss_type: mod

# Architecture
model: fm_base_12e_12d_swiglu_nobias
patch_size: 16
input_size: 224
dtype: bfloat16
tokenizer_path: "fourm/utils/tokenizer/trained/text_tokenizer_4m_wordpiece_30k.json"

# Train
epochs: -1
total_tokens: 100 # in Billions
opt: adamw
blr: 0.0001 # this is base_lr = 1e-4, lr = base_lr * batch_size / 256
min_blr: 0.
warmup_epochs: -1
warmup_tokens: 10 # in billions
batch_size: 128 # 128 x 32 = 4096

# Data
data_config: "cfgs/default/4m/data/cc12m/specialized/mix_caption_bias_a0.2_a1.0.yaml"
s3_data_endpoint: "/path/to/endpoint" # Change me
eval_freq: 1
fixed_eval: True
epoch_size: 10_000_000 # Number of samples per "epoch"

# Saving
save_ckpt_freq: 1 # Change if needed
output_dir: 'output/auto' # Change if needed

# Wandb
log_wandb: False # Set to True to log to Weights & Biases
wandb_project: '4m-train'
wandb_entity: null # Change if needed
wandb_run_name: auto # Change if needed