Skip to content

worldmodel: limit to planar pose conditioning#33

Merged
YassineYousfi merged 3 commits into
mainfrom
planar-pose-only
Jul 16, 2026
Merged

worldmodel: limit to planar pose conditioning#33
YassineYousfi merged 3 commits into
mainfrom
planar-pose-only

Conversation

@rkdune

@rkdune rkdune commented Jul 11, 2026

Copy link
Copy Markdown
  • add an opt-in pose-conditioning config flag
  • retain XY translation and yaw while masking Z, roll, and pitch
  • preserve caller-owned pose tensors during model forward

Comment thread torchtitan/experiments/worldmodel/model.py Outdated
rkdune added 2 commits July 16, 2026 14:23
* add an opt-in
  pose-conditioning config flag
* retain XY translation and yaw while masking Z, roll, and pitch
* preserve caller-owned pose
  tensors during model forward
@rkdune
rkdune force-pushed the planar-pose-only branch from a0df25e to 096b5c4 Compare July 16, 2026 21:56
@rkdune

rkdune commented Jul 16, 2026

Copy link
Copy Markdown
Author

Seems like failing lint check is due to an unrelated broken badge in the README, but out of scope for this PR.

@YassineYousfi
YassineYousfi merged commit 5efa6a4 into main Jul 16, 2026
4 of 5 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants