File tree Expand file tree Collapse file tree 1 file changed +26
-0
lines changed Expand file tree Collapse file tree 1 file changed +26
-0
lines changed Original file line number Diff line number Diff line change @@ -1637,4 +1637,30 @@ audio = model.autoencoder.decode(codes)[0].cpu()
16371637torchaudio.save("sample.wav", audio, model.autoencoder.sampling_rate)
16381638` ,
16391639] ;
1640+
1641+ export const smolvla = ( model : ModelData ) : string [ ] => [
1642+ `# Clone the LeRobot repository and create the environment
1643+ git clone https://github.com/huggingface/lerobot.git
1644+ cd lerobot
1645+ conda create -y -n lerobot python=3.10
1646+ conda activate lerobot
1647+
1648+ # Install ffmpeg (required for video processing)
1649+ conda install ffmpeg=7.1.1 -c conda-forge
1650+
1651+ # Install LeRobot with the SmolVLA extra dependencies
1652+ pip install -e ".[smolvla]"
1653+
1654+ # Launch finetuning on your dataset
1655+ cd lerobot && python lerobot/scripts/train.py \\
1656+ --policy.path=lerobot/smolvla_base \\
1657+ --dataset.repo_id=${ model . id } \\
1658+ --batch_size=64 \\
1659+ --steps=20000 \\
1660+ --output_dir=outputs/train/my_smolvla \\
1661+ --job_name=my_smolvla_training \\
1662+ --policy.device=cuda \\
1663+ --wandb.enable=true` ,
1664+ ] ;
1665+
16401666//#endregion
You can’t perform that action at this time.
0 commit comments