Update README.md

README.md

@@ -6,6 +6,45 @@ datasets:
 tags:
 - math
 ---
+
+### WIP
+
+## Usage:
+
+```
+import torch
+from peft import PeftModel
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
+
+model_name = "microsoft/phi-1_5"
+adapters_name = 'aloobun/phi_mini_math23k_v1'
+
+model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+    load_in_4bit=True,
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
+    quantization_config=BitsAndBytesConfig(
+        load_in_4bit=True,
+        bnb_4bit_compute_dtype=torch.bfloat16,
+        bnb_4bit_use_double_quant=True,
+        bnb_4bit_quant_type='nf4'
+    ),
+)
+model = PeftModel.from_pretrained(model, adapters_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+```
+
+```
+prompt = "What is the largest two-digit integer whose digits are distinct and form a geometric sequence?"
+formatted_prompt = (
+    f"### Instruction: {prompt} ### Response:"
+)
+inputs = tokenizer(formatted_prompt, return_tensors="pt").to("cuda:0")
+outputs = model.generate(inputs=inputs.input_ids, max_new_tokens=1048)
+print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
 ## Training procedure