Flatten checkpoint to 0200000.pth.tar; place via --local-dir on download

README.md

@@ -15,21 +15,19 @@ Aerial Navigation World Model checkpoint released with the paper
 
 Dataset: [EmbodiedCity/ANWM-Dataset](https://huggingface.co/datasets/EmbodiedCity/ANWM-Dataset)
 
-## Checkpoint
+## Files
 
 | File | Steps | Notes |
 |---|---:|---|
-| `logs/anwm_cdit_airvln/checkpoints/0200000.pth.tar` | 200,000 | Paper / released model (`config/anwm.yaml`) |
+| `0200000.pth.tar` | 200,000 | Paper / released model (`config/anwm.yaml`) |
 
 ## Quick start
 
-From the ANWM code repo root:
+From the ANWM code repo root, download into the path expected by `infer.py`:
 
 ```bash
-# download into the path expected by infer.py / check_environment.py
-huggingface-cli download EmbodiedCity/ANWM \
-  logs/anwm_cdit_airvln/checkpoints/0200000.pth.tar \
-  --local-dir .
+huggingface-cli download EmbodiedCity/ANWM 0200000.pth.tar \
+  --local-dir logs/anwm_cdit_airvln/checkpoints
 
 # optional check
 python scripts/check_environment.py --require-checkpoint
@@ -43,7 +41,7 @@ torchrun --standalone --nproc_per_node=1 infer.py \
   --rollout_fps_values 1,4
 ```
 
-The downloaded layout matches the documented local path:
+This places the weight at:
 
 ```text
 logs/anwm_cdit_airvln/checkpoints/0200000.pth.tar