BitDance_B_4x/transformer/config.json

{
  "_class_name": "BitDanceImageNetTransformer",
  "_diffusers_version": "0.36.0",
  "architecture": "BitDance-B",
  "parallel_num": 4,
  "resolution": 256,
  "down_size": 16,
  "latent_dim": 32,
  "num_classes": 1000,
  "source_checkpoint": "/data/projects/BitDance/models/shallowdream204/BitDance-ImageNet/BitDance_B_4x.pt",
  "source_key": "ema",
  "runtime_impl": "model_parallel.py",
  "parallel_mode": "patch",
  "time_schedule": "logit_normal",
  "time_shift": 1.0,
  "p_std": 1.0,
  "p_mean": 0.0
}