config.json

{
  "architectures": [
    "CompoDiffModel"
  ],
  "embed_dim": 768,
  "model_depth": 12,
  "model_dim": 64,
  "model_heads": 16,
  "model_type": "CompoDiff",
  "timesteps": 1000,
  "torch_dtype": "float32",
  "transformers_version": "4.25.1"
}