last-checkpoint/config.json

{
  "architectures": [
    "ControlTransformerWrapperPTM"
  ],
  "control_dim": 1,
  "embed_dim": 768,
  "grid_size": 48,
  "intermediate_dim": 3072,
  "nhead": 4,
  "nlayer": 4,
  "ntarget": 1,
  "torch_dtype": "float32",
  "transformers_version": "4.52.4"
}