Update README.md

README.md

@@ -62,8 +62,8 @@ class FormatPrompt_QA_with_citation():
         assert all([ item in instance.keys()  for item in self.inputs ]), logging.info(f"instance must have {self.inputs}!")
         
         prompt = f"""<s> [INST] Context: {str(instance["context"])}\
-        Question: {str(instance["question"])} [/INST]
-        Answer: """
+        Question: {str(instance["question"])} 
+        Answer: [/INST]"""
 
         if ('answer' in instance):
             if ('citation' in instance):
@@ -82,19 +82,20 @@ formatting_func = FormatPrompt_context_QA()
 
 # pull model from huggingface
 model, tokenizer = FastLanguageModel.from_pretrained(
-    model_name = "jingwang/mistral_context_qa",
+    model_name = "jingwang/mistral_qa_citation",
     max_seq_length = 2048,
     dtype = None,
     load_in_4bit = True,
 )
 
 
+# inference
 FastLanguageModel.for_inference(model)
 
 example = {'context': 'John Gadsby Chapman , The Baptism of Pocahontas (1840). A copy is on display in the Rotunda of the United States Capitol . During her stay at Henricus, Pocahontas met John Rolfe. Rolfe\'s English-born wife Sarah Hacker and child Bermuda had died on the way to Virginia after the wreck of the ship Sea Venture on the Summer Isles, now known as Bermuda. He established the Virginia plantation Varina Farms , where he cultivated a new strain of tobacco . Rolfe was a pious man and agonized over the potential moral repercussions of marrying a heathen, though in fact Pocahontas had accepted the Christian faith and taken the baptismal name Rebecca. In a long letter to the governor requesting permission to wed her, he expressed his love for Pocahontas and his belief that he would be saving her soul. He wrote that he was: motivated not by the unbridled desire of carnal affection, but for the good of this plantation, for the honor of our country, for the Glory of God, for my own salvation... namely Pocahontas, to whom my hearty and best thoughts are, and have been a long time so entangled, and enthralled in so intricate a labyrinth that I was even a-wearied to unwind myself thereout. [41] The couple were married on April 5, 1614, by chaplain Richard Buck , probably at Jamestown. For two years they lived at Varina Farms, across the James River from Henricus. Their son Thomas was born in January 1615. [42] The marriage created a climate of peace between the Jamestown colonists and Powhatan\'s tribes; it endured for eight years as the "Peace of Pocahontas". [43] In 1615, Ralph Hamor wrote, "Since the wedding we have had friendly commerce and trade not only with Powhatan but also with his subjects round about us." [44] The marriage was controversial in the British court at the time because "a commoner" had "the audacity" to marry a "princess." [45] [46]',
   'question': 'Who did Pocahontas marry?',
-  'answer': 'Pocahontas married John Rolfe',
-  'citation': 'The couple were married on April 5, 1614, by chaplain Richard Buck , probably at Jamestown.'
+  #'answer': 'Pocahontas married John Rolfe',
+  #'citation': 'The couple were married on April 5, 1614, by chaplain Richard Buck , probably at Jamestown.'
 }
 
 
@@ -105,7 +106,7 @@ input_length = inputs.input_ids.shape[-1]
 with torch.no_grad():
   output = model.generate(**inputs,
                           do_sample=False,
-                          temperature=0.1,
+                          temperature=0.5,
                           max_new_tokens=1024,
                           pad_token_id=tokenizer.eos_token_id,
                           use_cache=False,
@@ -116,11 +117,9 @@ with torch.no_grad():
                   )
   print(response)
 
-
-
 ```
 
 ```
->> {"answer": "Pocahontas married John Rolfe", "citation": "Rolfe's English-born wife Sarah Hacker and child Bermuda had died on the way to Virginia after the wreck of the ship Sea Venture on the Summer Isles, now known as Bermuda. He established the Virginia plantation Varina Farms , where he cultivated a new strain of tobacco . Rolfe was a pious man and agonized over the potential moral repercussions of marrying a heathen, though in fact Pocahontas had accepted the Christian"}
+>> {"answer": "Pocahontas married John Rolfe", "citation": "In a long letter to the governor requesting permission to wed her, he expressed his love for Pocahontas and his belief that he would be saving her soul. He wrote that he was: motivated not by the unbridled desire of carnal affection, but for the good of this plantation, for the honor of our country, for the Glory of God, for my own salvation... namely Pocahontas"}
 
 ```