Update README.md

README.md

@@ -14,27 +14,28 @@ This is a fine-tuned version of LLAMA2 trained (7b) on spider, sql-create-contex
 
 To initialize the model:
 
-bnb_config = BitsAndBytesConfig(
+
+    bnb_config = BitsAndBytesConfig(
     load_in_4bit=use_4bit,
     bnb_4bit_quant_type=bnb_4bit_quant_type,
     bnb_4bit_compute_dtype=compute_dtype,
     bnb_4bit_use_double_quant=use_nested_quant,
 )
 
-model = AutoModelForCausalLM.from_pretrained(
+    model = AutoModelForCausalLM.from_pretrained(
     model_name,
     quantization_config=bnb_config,
     device_map=device_map,
     trust_remote_code=True
 )
-    
+
     
 Use the tokenizer:
 
     
-tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-tokenizer.pad_token = tokenizer.eos_token
-tokenizer.padding_side = "right"
+    tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+    tokenizer.pad_token = tokenizer.eos_token
+    tokenizer.padding_side = "right"
 
 To get the prompt:
     dataset = dataset.map(