Praxis-1 Base

The Praxis-1 base model before embodiment-specific post-training: the chunk-causal mid-training checkpoint (P-base r13, step 526,120), trained on a mix of robot and human video across 18 embodiments. It has no per-embodiment action heads, so it does not run as a policy as is. Use it as the starting point for fine-tuning on a new rig.

Model 27.2B-parameter MoE video-diffusion policy: chunk-causal trunk + 0.3B action expert
Frozen dependencies (not in this repo) Wan2.2 VAE, Qwen/Qwen3.5-2B text encoder
Embodiment heads none ("embodiment": "base")
Control 15 Hz, 24-step action chunks
Weights bf16, 54.4 GB, 17 safetensors shards (model.safetensors.index.json)

Before fine-tuning: add the heads of your rig

The checkpoint has the video backbone and the shared action-expert trunk. It leaves out the six tensors that depend on the action and proprio size of a rig:

  • action_expert.action_encoder.w1.{weight,bias}
  • action_expert.embed_proprio.{weight,bias}
  • action_expert.head.{weight,bias}

config.json lists their shapes under heads. Make a checkpoint of your rig with scripts/init_heads.py in runwayml/praxis-1-oss, then fine-tune it as described in praxis/train/README.md, "Fine-tune the base checkpoint" (configs/finetune_base.yaml). The base config has no action_space, normalization or pose_frame; your rig's checkpoint sets them.

Files

  • config.json: praxis model config (model_type: praxis, format_version: 2)
  • model-000NN-of-00017.safetensors + model.safetensors.index.json: sharded weights
  • SHA256SUMS: checksums for every file above

Usage

from huggingface_hub import snapshot_download

path = snapshot_download("runwayml/praxis-1-base")

Then add heads and fine-tune with runwayml/praxis-1-oss.

Related

License

Apache License 2.0. The frozen dependencies (Wan2.2 VAE, Qwen/Qwen3.5-2B) are distributed separately under their own licenses.

Downloads last month
38
Safetensors
Model size
27B params
Tensor type
F32
·
BF16
·
Video Preview
loading

Model tree for runwayml/praxis-1-base

Finetunes
3 models

Collection including runwayml/praxis-1-base