Hugging Face's logo

plugyawn
/
rae-dit-s-ep14-diffusers

Unconditional Image Generation
Diffusers
Safetensors
RAEDiTPipeline
rae
rae-dit
diffusion-transformer
imagenet-256
arxiv:2510.11690
Model card Files
xet
 Community
plugyawn commited on
Commit
c4c58dd
·
verified ·
1 Parent(s): 84f7e37

Fix RAEDiT Stage-2 positional embedding export

Browse files
Files changed (3) hide show
  1. conversion_metadata.json +1 -1
  2. transformer/config.json +1 -1
  3. transformer/diffusion_pytorch_model.safetensors +1 -1
conversion_metadata.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "source": {
3
  "weights_repo_or_path": "nyu-visionx/RAE-collections",
4
- "config_repo_or_path": "/home/ubuntu/RAE",
5
  "config_path": "configs/stage2/training/ImageNet256/DiTDH-S_DINOv2-B.yaml",
6
  "vae_model_name_or_path": "nyu-visionx/RAE-dinov2-wReg-base-ViTXL-n08"
7
  },
 
1
  {
2
  "source": {
3
  "weights_repo_or_path": "nyu-visionx/RAE-collections",
4
+ "config_repo_or_path": "/home/ubuntu/rae-upstream-configs",
5
  "config_path": "configs/stage2/training/ImageNet256/DiTDH-S_DINOv2-B.yaml",
6
  "vae_model_name_or_path": "nyu-visionx/RAE-dinov2-wReg-base-ViTXL-n08"
7
  },
transformer/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "RAEDiT2DModel",
3
  "_diffusers_version": "0.38.0.dev0",
4
- "_name_or_path": "/home/ubuntu/rae-dit-s-ep14-diffusers/transformer",
5
  "class_dropout_prob": 0.1,
6
  "depth": [
7
  12,
 
1
  {
2
  "_class_name": "RAEDiT2DModel",
3
  "_diffusers_version": "0.38.0.dev0",
4
+ "_name_or_path": "/tmp/tmppwg9zch4/transformer",
5
  "class_dropout_prob": 0.1,
6
  "depth": [
7
  12,
transformer/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:05b70b7b5a1f65c2c3d70ef9886c27a15f2eb71c2894053abc5dcde59b14d5e1
3
  size 786240304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5d5b95a07bed32ef39f1958e3dcf436bccda3c92b4ed9978c3105b25f4ace4b
3
  size 786240304