Update README.md

README.md

@@ -5,6 +5,7 @@
 To use this model, first install the custom package:
 
 ```bash
+# Install required packages
 !pip install bitsandbytes accelerate
 !pip install git+https://huggingface.co/CognitivessAI/cognitivess
 ```
@@ -12,6 +13,7 @@ To use this model, first install the custom package:
 Then, you can use the model like this:
 
 ```python
+
 # Import necessary libraries
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
 import torch
@@ -39,6 +41,7 @@ config.quantization_config = quantization_config
 model = CognitivessForCausalLM.from_pretrained(
     "CognitivessAI/cognitivess",
     config=config,
+    quantization_config=quantization_config,
     device_map="auto"
 )