Update README.md

README.md

@@ -315,4 +315,32 @@ def generate(text):
 # Now you can simply call the generate function with an English text you want to translate:
 generate("I'm super excited about this Norwegian NORA model! Can it translate these sentences?")
 # > this should output: 'Jeg er super spent på denne norske NORA modellen! Kan den oversette disse setningene?'
+```
+
+## Example usage on a GPU with ~16GB VRAM (try for yourself [in Google Colab](https://colab.research.google.com/drive/1AQgJ8lN-SNOqkUKj4xpQI5rr0R7V2Xzy?usp=sharing))
+Install bitsandbytes if you want to load in 8bit
+
+```bash
+pip install bitsandbytes
+pip install accelerate
+```
+
+
+```python
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+tokenizer = AutoTokenizer.from_pretrained(
+    "norallm/normistral-7b-scratch"
+)
+
+# This setup needs about 8gb VRAM
+# Setting `load_in_8bit=False` -> 15gb VRAM
+# Using `torch.float32` and `load_in_8bit=False` -> 21gb VRAM
+model = AutoModelForCausalLM.from_pretrained(
+    "norallm/normistral-7b-scratch",
+    device_map='auto',
+    load_in_8bit=True,
+    torch_dtype=torch.bfloat16
+)
 ```