From 399f8ed56f40df0919208d1ffdee64a31a1b22c8 Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Fri, 15 Dec 2023 16:35:24 +1100
Subject: [PATCH] Update Colab to CUDA 12.1

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 44abff8e61..b93dcb748f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -65,7 +65,7 @@ kaggle = [
 	"unsloth[huggingface]",
 ]
 colab = [
-	"unsloth[cu118]",
+	"unsloth[cu121]",
 ]
 cu118_ampere = [
 	"unsloth[huggingface]",
@@ -82,7 +82,7 @@ cu121_ampere = [
 	"flash-attn",
 ]
 colab_ampere = [
-	"unsloth[cu118]",
+	"unsloth[cu121]",
 	"ninja",
 	"flash-attn",
 ]