config.json

{
  "hidden_size": 384,
  "img_size": 32,
  "num_blocks": 8,
  "num_channels": 4,
  "num_genres": 42,
  "num_heads": 6,
  "num_styles": 137,
  "patch_size": 2
}