Skip to content

Commit 00b8578

Browse files
Update model-libraries-snippets.ts
1 parent 85e2ce1 commit 00b8578

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1638,8 +1638,8 @@ torchaudio.save("sample.wav", audio, model.autoencoder.sampling_rate)
16381638
`,
16391639
];
16401640

1641-
export const smolvla = (model: ModelData): string[] => [
1642-
`# Clone the LeRobot repository and create the environment
1641+
export const smolvla = (): string[] => [
1642+
`# Clone the LeRobot repository and create the environment
16431643
git clone https://github.com/huggingface/lerobot.git
16441644
cd lerobot
16451645
conda create -y -n lerobot python=3.10
@@ -1649,12 +1649,12 @@ conda activate lerobot
16491649
conda install ffmpeg=7.1.1 -c conda-forge
16501650
16511651
# Install LeRobot with the SmolVLA extra dependencies
1652-
pip install -e ".[${model.id}]"
1652+
pip install -e ".[smolvla]"
16531653
16541654
# Launch finetuning on your dataset
16551655
cd lerobot && python lerobot/scripts/train.py \\
16561656
--policy.path=lerobot/smolvla_base \\
1657-
--dataset.repo_id=${model.id} \\
1657+
--dataset.repo_id=lerobot/svla_so101_pickplace \\
16581658
--batch_size=64 \\
16591659
--steps=20000 \\
16601660
--output_dir=outputs/train/my_smolvla \\

0 commit comments

Comments
 (0)