upload checkpoints

README.md

@@ -63,8 +63,8 @@ python inference.py \
   --image test_images/test_1.jpg \
   --output outputs/result.png \
   --seed 123 \
-  --steps 30 \
-  --guidance-scale 5.0 \
+  --steps 50 \
+  --guidance-scale 4.0 \
   --basesize 1024
 ```
 
@@ -79,13 +79,12 @@ python inference.py \
 | `--image` | str | None | Input image path (required for editing, omit for T2I) |
 | `--output` | str | `example.png` | Output image path |
 | `--steps` | int | 50 | Denoising steps |
-| `--guidance-scale` | float | 5.0 | Classifier-free guidance scale |
+| `--guidance-scale` | float | 4.0 | Classifier-free guidance scale |
 | `--seed` | int | 42 | Random seed for reproducibility |
 | `--neg-prompt` | str | `""` | Negative prompt |
 | `--basesize` | int | 1024 | Bucket base size for input image resizing (256/512/768/1024) |
 | `--config` | str | auto | Config path; defaults to `<ckpt-root>/infer_config.py` |
 | `--rewrite-prompt` | flag | off | Enable LLM-based prompt rewriting |
-| `--rewrite-model` | str | `gpt-5` | Model name for prompt rewriting |
 | `--hsdp-shard-dim` | int | 1 | FSDP shard dimension for multi-GPU (set to GPU count) |
 
 

config.json

@@ -0,0 +1,7 @@
+{
+    "root": "JoyAI-Image-Edit",
+    "transformer_ckpt": "JoyAI-Image-Edit/transformer/transformer.pth",
+    "vae_ckpt": "JoyAI-Image-Edit/vae/Wan2.1_VAE.pth",
+    "text_encoder_ckpt": "JoyAI-Image-Edit/JoyAI-Image-Und"
+  }
+