Update README.md

README.md

@@ -4,7 +4,19 @@ library_name: peft
 ---
 How to use :
 ```python
+  %%capture
+import torch
+major_version, minor_version = torch.cuda.get_device_capability()
+# Must install separately since Colab has torch 2.2.1, which breaks packages
 !pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
+if major_version >= 8:
+    # Use this for new GPUs like Ampere, Hopper GPUs (RTX 30xx, RTX 40xx, A100, H100, L40)
+    !pip install --no-deps packaging ninja einops flash-attn xformers trl peft accelerate bitsandbytes
+else:
+    # Use this for older GPUs (V100, Tesla T4, RTX 20xx)
+    !pip install --no-deps xformers trl peft accelerate bitsandbytes
+pass
+
 import torch
 from transformers import TextStreamer
 from unsloth import FastLanguageModel