Update install instruction (#1602)

Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>
neuralmagic · Jun 9, 2023 · 07e1459 · 07e1459
1 parent 27c6276
commit 07e1459
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/integrations/ultralytics-yolov5/tutorials/sparse-transfer-learning.md b/integrations/ultralytics-yolov5/tutorials/sparse-transfer-learning.md
@@ -31,7 +31,7 @@ SparseZoo contains pre-sparsified checkpoints of each YOLOv5 model. These models
 Install via `pip`:
 
 ```
-pip install sparseml[torchvision]
+pip install "sparseml[yolov5]"
 ```
 
 ## Table of Contents