Instructions to use WetLabRoboData/diffusion-cap-finetuned with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- LeRobot
How to use WetLabRoboData/diffusion-cap-finetuned with LeRobot:
# No code snippets available yet for this library. # To use this model, check the repository files and the library's documentation. # Want to help? PRs adding snippets are welcome at: # https://github.com/huggingface/huggingface.js
- Notebooks
- Google Colab
- Kaggle
File size: 1,605 Bytes
e43b78f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 | ---
license: apache-2.0
library_name: lerobot
tags:
- lerobot
- diffusion-policy
- imitation-learning
- robotics
- ur3e
- bimanual
datasets:
- WetLabRoboData/lerobot-data-cap
pipeline_tag: robotics
base_model: WetLabRoboData/diffusion-multitask_12task_mix-multitask
---
# diffusion-cap-finetuned
LeRobot diffusion policy for the **cap** task.
| Field | Value |
|--------------------|-------|
| Policy family | diffusion (LeRobot) |
| Variant | Finetuned (multitask pretrain, then finetuned on this task's finetune pool) |
| Training dataset | [`WetLabRoboData/lerobot-data-cap`](https://huggingface.co/datasets/WetLabRoboData/lerobot-data-cap) |
| Base model | [`WetLabRoboData/diffusion-multitask_12task_mix-multitask`](https://huggingface.co/WetLabRoboData/diffusion-multitask_12task_mix-multitask) |
| Target task | cap |
| Eval episodes | 20 |
| Eval successes | 6 / 20 |
| Robot | UR3e bimanual (3 cameras) |
## Evaluation
Rollout videos + per-episode outcomes: [`WetLabRoboData/eval-diffusion-cap-finetuned`](https://huggingface.co/datasets/WetLabRoboData/eval-diffusion-cap-finetuned).
## Load and deploy
```python
from lerobot.policies.diffusion.modeling_diffusion import DiffusionPolicy
policy = DiffusionPolicy.from_pretrained("WetLabRoboData/diffusion-cap-finetuned")
```
## Provenance
Reorganized from `WetLabRoboData/lerobot-data-rama-lbm_finetune_cap` on 2026-10-04. The archived
training outputs (checkpoints, train_config.json, wandb/) are preserved in
the `old/` subfolder of that source repo for traceability.
|