Update README.md

README.md

@@ -21,19 +21,22 @@ Requirements:
 Import this model using:
 
 <pre>
-import torch
-from peft import PeftModel, PeftConfig
-from transformers import AutoModelForCausalLM, AutoTokenizer
+<code>
+<span style="color: #0000FF;">import</span> torch
+<span style="color: #0000FF;">from</span> peft <span style="color: #0000FF;">import</span> PeftModel, PeftConfig
+<span style="color: #0000FF;">from</span> transformers <span style="color: #0000FF;">import</span> AutoModelForCausalLM, AutoTokenizer
 
-peft_model_id = "AhmedBou/databricks-dolly-v2-3b_on_NCSS"
+peft_model_id = "<span style="color: #A31515;">"AhmedBou/databricks-dolly-v2-3b_on_NCSS"</span>
 config = PeftConfig.from_pretrained(peft_model_id)
-model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=True, device_map='auto')
+model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, return_dict=<span style="color: #0000FF;">True</span>, load_in_8bit=<span style="color: #0000FF;">True</span>, device_map=<span style="color: #0000FF;">'auto'</span>)
 tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 
-# Load the Lora model
+<span style="color: #808080;"># Load the Lora model</span>
 model = PeftModel.from_pretrained(model, peft_model_id)
+</code>
 </pre>
 
+
 Inference using:
 
 <pre>