Upload folder using huggingface_hub

README.md

@@ -34,7 +34,6 @@ The transformers version must be **between 4.57 and 4.58**; otherwise, incorrect
 | `torch` | >= 2.8 | PyTorch |
 | `transformers` | >= 4.57.0, < 4.58.0 | Text encoder |
 | `torchvison` | - |Image process|
-| `einops` | - |Tensor manipulation|
 
 ### Install the [Pull Request](https://github.com/huggingface/diffusers/pull/13444]) of JoyAI-Image-Edit of diffusers
 ```bash
@@ -43,7 +42,7 @@ pip install git+https://github.com/huggingface/diffusers.git@refs/pull/13444/hea
 
 ### Or install from this repo (PR will merge to diffusers main branch soon)
 ```bash
-pip install torch==2.8 transformers==4.57.6 torchvision einops
+pip install torch==2.8 transformers==4.57.6 torchvision
 
 pip install git+https://github.com/Moran232/diffusers.git@joyimage_edit
 ```

transformer/config.json

@@ -1,22 +1,19 @@
 {
   "_class_name": "JoyImageEditTransformer3DModel",
   "_diffusers_version": "0.38.0.dev0",
-  "attn_backend": "torch_spda",
-  "dit_modulation_type": "wanx",
-  "enable_activation_checkpointing": false,
   "heads_num": 32,
   "hidden_size": 4096,
   "in_channels": 16,
-  "is_repa": false,
   "mlp_width_ratio": 4.0,
   "mm_double_blocks_depth": 40,
+  "num_attention_heads": 24,
+  "num_layers": 20,
   "out_channels": 16,
   "patch_size": [
     1,
     2,
     2
   ],
-  "repa_layer": 13,
   "rope_dim_list": [
     16,
     56,
@@ -24,6 +21,7 @@
   ],
   "rope_theta": 10000,
   "rope_type": "rope",
+  "text_dim": 4096,
   "text_states_dim": 4096,
-  "unpatchify_new": true
+  "theta": 256
 }

transformer/diffusion_pytorch_model-00001-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e753a3081d3effb6138bd33b46cf1a672791372c9b49acb3c08746c8d57530e5
-size 4869559736
+oid sha256:adcb100cae21cbdc71d75cd26de468531bd1c8e49e718e3d377530a0034ca33a
+size 4970222408

transformer/diffusion_pytorch_model-00002-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d5e50771f9ebbc80f5504fe2ae49e66c34eb04c070c28f8baa90ece1a29eb2cb
-size 4967584552
+oid sha256:6c76310e69fa291b954280bc728a1a2de95533455ae01f6e6b6aaa58711b10e8
+size 4967585880

transformer/diffusion_pytorch_model-00003-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9507d7e97c2dc4e035590617536670763bbd198ede468a706ca9fa1ed05f0a13
-size 4967609344
+oid sha256:845645c96a41269d02e6b02de953d3c58ff2741f33a326823feaccdd65f7937f
+size 4866946400

transformer/diffusion_pytorch_model-00004-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cea48ca97bdda33daa37babe6bb1970f58987dcf9ace045bf17df83282363535
-size 4967585592
+oid sha256:722c95c3b39233f598dc6ffef589ecb411e8531ebfe4241c0d036f93c74e1bf8
+size 4967584992

transformer/diffusion_pytorch_model-00005-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4bc45fd90800e75c7096e295e50cb574080dd5d34cc5a81d9a63335220c840d7
-size 4967584672
+oid sha256:0a8e110439a119ad132e1d56ba137b63c48c232ffbd974572c37140f2becbd8c
+size 4967609680

transformer/diffusion_pytorch_model-00006-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6212a52be1e89724d30681f28fdd35a3b8c4b4b98cb44155f39d58b3171abd54
-size 4967609344
+oid sha256:ab5e2cc1b6e15c57ca9fe8126d81141b5a52e9a19a1cbbff8afb7c9e130cee7e
+size 4967585000

transformer/diffusion_pytorch_model-00007-of-00007.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:69ab8a991b62b7de87b088f8439b100107d6ebf45f3dcefc82058b4d2d524e0f
-size 2819920088
+oid sha256:b359f79a76e35db3b28be8470a3be1af8be23ec396861e2c641697538ffb8c55
+size 2819921264