README.md

metadata

license: mit
tags:
  - diffusion-policy
  - lerobot
  - safetensors
  - so101

Diffusion Policy: so101-block-horizontal-layComb12

Pretrained Diffusion Policy trained using the LeRobot framework.

📦 Checkpoint Info

• Format: safetensors
• Model: Diffusion policy
• Trained on: so101-block-horizontal-layComb12 dataset
• Steps: 25,000
• Final loss: ~0.013
• Config: train_config.json, config.json

🔧 How to Load

from huggingface_hub import hf_hub_download
import safetensors.torch as storch
import torch

# Load model weights
model_path = hf_hub_download("tshiamor/diffussion_so101-block-horizontal-layComb12", "model.safetensors")
state_dict = storch.load_file(model_path)

# You can now load state_dict into your model architecture