Update README.md

README.md

@@ -48,9 +48,19 @@ Generating a report for a chest X-ray image using this model is straightforward.
 ### 1. Install Necessary Libraries
 
 ```bash
-pip install "unsloth[colab-new] @ git+[https://github.com/unslothai/unsloth.git](https://github.com/unslothai/unsloth.git)"
-pip install --no-deps transformers trl peft accelerate bitsandbytes
-pip install Pillow # For image processing
+
+%%capture
+import os, re
+if "COLAB_" not in "".join(os.environ.keys()):
+    !pip install unsloth
+else:
+    import torch; v = re.match(r"[0-9\\.]{3,}", str(torch.__version__)).group(0)
+    xformers = "xformers==" + ("0.0.32.post2" if v == "2.8.0" else "0.0.29.post3")
+    !pip install --no-deps bitsandbytes accelerate {xformers} peft trl triton cut_cross_entropy unsloth_zoo
+    !pip install sentencepiece protobuf "datasets>=3.4.1,<4.0.0" "huggingface_hub>=0.34.0" hf_transfer
+    !pip install --no-deps unsloth
+!pip install transformers==4.55.4
+!pip install --no-deps trl==0.22.2
 ```
 
 ### 2. Run Inference with Python