Update README.md

README.md

@@ -47,15 +47,19 @@ model_id = "NurtureAI/llama-2-7b-int4-gptq-python"
 tokenizer = AutoTokenizer.from_pretrained(hf_model_repo)
 model = AutoModelForCausalLM.from_pretrained(
   hf_model_repo, load_in_4bit=True,
-  torch_dtype=torch.float16, device_map=device_map)
+  torch_dtype=torch.float16, device_map=device_map,
+)
 instruction="Write a Python function to display the first and last elements of a list."
 input=""
 prompt = f"""### Instruction:
 Use the Task below and the Input given to write the Response, which is a programming code that can solve the Task.
+
 ### Task:
 {instruction}
+
 ### Input:
 {input}
+
 ### Response:
 """
 input_ids = tokenizer(prompt, return_tensors="pt", truncation=True).input_ids.cuda()