Robotics
Transformers
Safetensors
citywalker
feature-extraction
navigation
waypoint-prediction
dinov2
custom_code
Instructions to use ai4ce/citywalker with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use ai4ce/citywalker with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("ai4ce/citywalker", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
README: drop stale load_obs_encoder() snippet, add trust_remote_code=True example
Browse files
README.md
CHANGED
|
@@ -17,7 +17,7 @@ waypoint-prediction model (CVPR 2025), trained on ~2000 hours of web-scale
|
|
| 17 |
urban walking and driving videos. This repo contains the converted weights
|
| 18 |
of `CityWalker_2000hr.ckpt` (originally a PyTorch Lightning checkpoint)
|
| 19 |
re-packaged as a `transformers.PreTrainedModel` so it can be loaded with
|
| 20 |
-
`AutoModel.from_pretrained`.
|
| 21 |
|
| 22 |
Upstream training dataset: [ai4ce/CityWalker](https://huggingface.co/datasets/ai4ce/CityWalker).
|
| 23 |
Our port (model wrapper + ckpt converter + benchmark integration) lives in
|
|
@@ -46,15 +46,18 @@ coords (B, 6, 2) ─► PolarEmbedding + Linear ─► goal token (B,
|
|
| 46 |
|
| 47 |
```python
|
| 48 |
from transformers import AutoModel
|
| 49 |
-
from wanderland_lab.models.citywalker import CityWalkerModel # registers AutoModel
|
| 50 |
|
| 51 |
-
model = AutoModel.from_pretrained(
|
| 52 |
-
|
|
|
|
|
|
|
| 53 |
model.eval()
|
| 54 |
```
|
| 55 |
|
| 56 |
-
The
|
| 57 |
-
|
|
|
|
|
|
|
| 58 |
|
| 59 |
## Inputs / Outputs
|
| 60 |
|
|
|
|
| 17 |
urban walking and driving videos. This repo contains the converted weights
|
| 18 |
of `CityWalker_2000hr.ckpt` (originally a PyTorch Lightning checkpoint)
|
| 19 |
re-packaged as a `transformers.PreTrainedModel` so it can be loaded with
|
| 20 |
+
`AutoModel.from_pretrained` directly.
|
| 21 |
|
| 22 |
Upstream training dataset: [ai4ce/CityWalker](https://huggingface.co/datasets/ai4ce/CityWalker).
|
| 23 |
Our port (model wrapper + ckpt converter + benchmark integration) lives in
|
|
|
|
| 46 |
|
| 47 |
```python
|
| 48 |
from transformers import AutoModel
|
|
|
|
| 49 |
|
| 50 |
+
model = AutoModel.from_pretrained(
|
| 51 |
+
"ai4ce/citywalker",
|
| 52 |
+
trust_remote_code=True,
|
| 53 |
+
)
|
| 54 |
model.eval()
|
| 55 |
```
|
| 56 |
|
| 57 |
+
The repo bundles `modeling_citywalker.py` and `configuration_citywalker.py`
|
| 58 |
+
under `auto_map`, so `trust_remote_code=True` is all you need — no need to
|
| 59 |
+
pip-install the wanderland-lab package. The DINOv2 backbone weights are
|
| 60 |
+
included in `model.safetensors`.
|
| 61 |
|
| 62 |
## Inputs / Outputs
|
| 63 |
|