Add colab link

README.md

@@ -20,6 +20,9 @@ This is a fine-tuned model from [Meta/Meta-Llama-3-8B](https://huggingface.co/Me
 The motivation behind this model was to fine-tune an LLM that is capable of understanding the nuances of the Drilling Operations and provide 24-hour summarizations based on the inputs from Daily Drilling Reports hourly activities.
 
 ## How to use
+### Sample Colab
+Here's a [Google colab notebook](https://colab.research.google.com/drive/10Txp14M-yeJG3hRAB8U2ydPrWFE1bypW?usp=sharing) where you can get started with using the model 
+
 ### Recommended template for DriLLM-Summarizer:
 ``` python
 TEMPLATE = """<|begin_of_text|>Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request.
@@ -92,11 +95,13 @@ If you are facing GPU constraints, you can try to load it with 8-bit quantizatio
 ``` python
 from transformers import BitsAndBytesConfig
 
-  pipeline = transformers.pipeline(
-      "text-generation", 
-      model=model_id, 
-      torch_dtype = torch.bfloat16, 
-      quantization_config=BitsAndBytesConfig(load_in_8bit=True), # 8-bit quantization, 
-      device_map="auto"
-  )
+pipeline = transformers.pipeline(
+    "text-generation", 
+    model=model_id, 
+    model_kwargs = {
+        "torch_dtype":  torch.bfloat16, 
+        "quantization_config": BitsAndBytesConfig(load_in_8bit=True), # Uncomment to use 8-bit quantization, 
+    },
+    device_map="auto"
+)
 ```