config.json

{
  "cond_drop_prob": 0.1,
  "device": "cuda",
  "frozen_clip": true,
  "heads": 8,
  "init_scale": 0.25,
  "init_with_modified_layer": true,
  "input_channels": 6,
  "layers": 12,
  "n_ctx": 2048,
  "output_channels": 12,
  "pointe_cache_dir": "cache/point_e_model",
  "time_token_cond": true,
  "token_cond": true,
  "width": 512
}