Update README.md

README.md

@@ -1,9 +1,15 @@
 ---
 library_name: peft
+language:
+- en
+pipeline_tag: text-generation
+tags:
+- QA
+- Objective QA
+- Subjective QA
 ---
 ## Training procedure
 
-
 The following `bitsandbytes` quantization config was used during training:
 - quant_method: QuantizationMethod.BITS_AND_BYTES
 - load_in_8bit: False
@@ -15,7 +21,51 @@ The following `bitsandbytes` quantization config was used during training:
 - bnb_4bit_quant_type: nf4
 - bnb_4bit_use_double_quant: False
 - bnb_4bit_compute_dtype: bfloat16
+  
 ### Framework versions
+- PEFT 0.4.0
 
+## Methods to Infer
+```
+from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, pipeline, TextStreamer
 
-- PEFT 0.4.0
+def stream(user_prompt):
+    runtimeFlag = "cuda:0"
+    inputs = tokenizer([user_prompt], return_tensors="pt").to(runtimeFlag)
+    streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=False)
+    _ = model.generate(**inputs, streamer=streamer, max_new_tokens=2046)
+
+def infer(user_prompt):
+    runtimeFlag = "cuda:0"
+    inputs = tokenizer([user_prompt], return_tensors="pt").to(runtimeFlag)
+    outputs = model.generate(**inputs, max_new_tokens=2046)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+def infer_pipeline(user_prompt):
+    runtimeFlag = "cuda:0"
+    pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=2046, return_full_text=False, device_map="auto",
+                    kwargs={'stop': ["###Human:"]})
+    return pipe(user_prompt)
+```
+
+### Sample Inputs:
+```
+Sys_OBJECTIVE = '''You are a chatbot, who is helping to curate datasets. When given an input context paragraph, you have to generate only one mcq question, 
+it's options and it's actual answer. You have to follow the given JSON format for generating the question, options and answer.
+Donot use words like "in this paragraph", "from the context" etc. The questions should be independent of any other question.
+'''
+
+Sys_SUBJECTIVE = '''You are a chatbot, who is helping to curate datasets. When given an input context paragraph, you have to generate only one subjective quesion,
+and it's actual answer. You have to follow the given JSON format for generating the question and answer.
+Donot use words like "in this paragraph", "from the context" etc. The questions should be independent of any other question.'''
+
+Prompt = '''And in the leadership styles it will be that is the is the there will be the changing into the leadership styles and in the leadership styles it will be that is the the approach will be for doing this type of the research which has been adopted in this paper is that is the degree of the correlation and its statistical significance between the self-assess leadership behavior and the 360 degree assessment of performance, evidence is presented showing that results vary in different context.'''
+ 
+Formatted_Prompt_OBJECTIVE = f"###Human: {Sys_OBJECTIVE}\nThe context is: {Prompt}\n###Assistant: "
+
+Formatted_Prompt_SUBJECTIVE = f"###Human: {Sys_SUBJECTIVE}\nThe context is: {Prompt}\n###Assistant: "
+
+- stream(Formatted_Prompt_OBJECTIVE)
+- infer_pipeline(Formatted_Prompt_OBJECTIVE)
+- infer_pipeline(Formatted_Prompt_SUBJECTIVE)
+```