diff --git a/Colab-Inference.ipynb b/Colab-Inference.ipynb index b565e4e1..b962c9b3 100644 --- a/Colab-Inference.ipynb +++ b/Colab-Inference.ipynb @@ -60,7 +60,7 @@ "\n", "pip install ipykernel\n", "\n", - "bash install.sh --device CU124 --source HF" + "bash install.sh --device CU126 --source HF" ] }, {