Update README.md

README.md

@@ -40,13 +40,15 @@ from transformers import pipeline
 # Chose the model inference precision
 dtype = torch.float16 # options are torch.float16, torch.bfloat16, torch.float32
 
-model = pipeline(model="VMware/flan-ul2-alpaca-lora",device_map = 'auto',torch_dtype=dtype )
+model = pipeline(
+    model = "VMware/flan-ul2-alpaca-lora",
+    device_map = 'auto',
+    torch_dtype=dtype
+)
 
 prompt = "YOUR PROMPT HERE"
 
 output = model(prompt, max_length=2048, do_sample=True)
-
-
 ```
 
 
@@ -60,14 +62,17 @@ from transformers import pipeline
 dtype = torch.float16 # options are torch.bfloat16, torch.float32
 model = pipeline(model="VMware/flan-ul2-alpaca-lora",device_map = 'auto',torch_dtype=dtype )
 
-prompt_template = "Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{instruction}\n\n### Response:"
-
-prompt = "YOUR PROMPT HERE"
+prompt_template = """
+Below is an instruction that describes a task. Write a response that appropriately completes the request.
 
-output = model(prompt_template.format(instruction= prompt), max_length=2048, do_sample=True)
+### Instruction:
+{instruction}
 
+### Response:"""
 
+prompt = "YOUR INSTRUCTION HERE"
 
+output = model(prompt_template.format(instruction=prompt), max_length=2048, do_sample=True)
 ```
 
 # Training Details
@@ -81,10 +86,6 @@ The model was trained on 3xV100 GPUs using PEFT-LORA and Deepspeed
 *  epochs = 3
 
 
-```
-
-
-
 # Limitations and Bias
 
 The model is based on a large and diverse dataset, but it may still have limitations and biases in certain areas. Some limitations include: