Update README.md

README.md

@@ -21,8 +21,17 @@ Refer to the [original model](https://huggingface.co/stabilityai/stablelm-base-a
 - total model size is only ~7 GB! (Assuming model size reduction similar to the dolly-v2-12b model)
 - this enables low-RAM loading, i.e. Colab :)
 
+
 ## Basic Usage
 
+You can use this model as a drop-in replacement in the notebook for the standard sharded models:
+
+<a href="https://colab.research.google.com/gist/pszemraj/4bd75aa3744f2a02a5c0ee499932b7eb/sharded-stablelm-testing-notebook.ipynb">
+  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
+</a>
+
+### Python
+
 Install/upgrade `transformers`, `accelerate`, and `bitsandbytes`. For this to work **you must have** `transformers>=4.28.0` and `bitsandbytes>0.37.2`.
 
 ```bash
@@ -39,8 +48,3 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 model = AutoModelForCausalLM.from_pretrained(model_name)
 ```
-You can also use this model as a drop-in replacement in the notebook for the standard sharded models:
-
-<a href="https://colab.research.google.com/gist/pszemraj/4bd75aa3744f2a02a5c0ee499932b7eb/sharded-stablelm-testing-notebook.ipynb">
-  <img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/>
-</a>