Final: Leplanner Controller/{pusht,reacher,tworooms}/controller.pt + density.pt

README.md

@@ -31,6 +31,7 @@ from huggingface_hub import hf_hub_download
 env = "reacher"  # or "pusht" / "tworooms"
 
 controller = hf_hub_download("nottygian/Leplanner", f"checkpoints/Leplanner Controller/{env}/controller.pt", repo_type="model")
+density   = hf_hub_download("nottygian/Leplanner", f"checkpoints/Leplanner Controller/{env}/density.pt", repo_type="model")
 world_model = hf_hub_download("nottygian/Leplanner", f"checkpoints/base_world_models/{env}/weights.pt", repo_type="model")
 ```
 
@@ -38,10 +39,16 @@ world_model = hf_hub_download("nottygian/Leplanner", f"checkpoints/base_world_mo
 
 ```
 checkpoints/
-├── Leplanner Controller/             # Trained controllers
-│   ├── pusht/controller.pt           # 98% success
-│   ├── reacher/controller.pt         # 100% success
-│   └── tworooms/controller.pt        # 100% success
+├── Leplanner Controller/             # Trained controllers + density
+│   ├── pusht/
+│   │   ├── controller.pt             # 98% success
+│   │   └── density.pt
+│   ├── reacher/
+│   │   ├── controller.pt             # 100% success
+│   │   └── density.pt
+│   └── tworooms/
+│       ├── controller.pt             # 100% success
+│       └── density.pt
 ├── base_world_models/                # Frozen LeWM world models
 │   ├── pusht/config.json + weights.pt
 │   ├── reacher/config.json + weights.pt