Gaaaavin commited on
Commit
1b5d169
·
verified ·
1 Parent(s): cec7d2d

README: drop stale load_obs_encoder() snippet, add trust_remote_code=True example

Browse files
Files changed (1) hide show
  1. README.md +9 -6
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("ai4ce/citywalker")
52
- model.load_obs_encoder() # fetches DINOv2 via torch.hub on first call
 
 
53
  model.eval()
54
  ```
55
 
56
- The DINOv2 backbone is not bundled with the weights to avoid redistributing
57
- Meta's pretrained checkpoint; `load_obs_encoder()` pulls it via `torch.hub`.
 
 
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