config.json

{
  "hidden_size": 640,
  "img_size": 32,
  "num_blocks": 12,
  "num_channels": 4,
  "num_genres": 42,
  "num_heads": 10,
  "num_styles": 137,
  "patch_size": 2
}