Skip to content

Commit 75517d1

Browse files
update model-libraries-snippets.ts
1 parent 0f9d592 commit 75517d1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

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

1641-
export const smolvla = (): string[] => [
1641+
export const smolvla = (model: ModelData): string[] => [
16421642
`# git clone https://github.com/huggingface/lerobot.git
16431643
cd lerobot
16441644
conda create -y -n lerobot python=3.10
@@ -1648,7 +1648,7 @@ conda activate lerobot
16481648
conda install ffmpeg=7.1.1 -c conda-forge
16491649
16501650
# Install LeRobot with the SmolVLA extra dependencies
1651-
pip install -e ".[smolvla]"
1651+
pip install -e ".[${model.id}]"
16521652
16531653
# Launch finetuning on your dataset
16541654
cd lerobot && python lerobot/scripts/train.py \\

0 commit comments

Comments
 (0)