Update README.md

README.md

@@ -12,7 +12,7 @@ pipeline_tag: question-answering
 
 <!-- Provide a quick summary of what the model is/does. -->
 
-This is an experimental product that can be used to create new LLM bassed on Chinese language. It has been generated using [Chinese-LLaMA-Alpaca](https://github.com/ymcui/Chinese-LLaMA-Alpaca)
+This is an experimental product that can be used to create new LLM bassed on Chinese language. It is created based on the [Chinese-LLaMA-Alpaca](https://github.com/ymcui/Chinese-LLaMA-Alpaca)
 ## Model Details
 
 ### Model Description
@@ -30,7 +30,7 @@ This is an experimental product that can be used to create new LLM bassed on Chi
 
 <!-- Provide the basic links for the model. -->
 
-- **Repository:** https://huggingface.co/BlueWhaleX/Chinese-Alpaca-COIG-49954-7B-HF
+- **Repository:** https://huggingface.co/BlueWhaleX/bwx-7B-HF
 
 ## Uses
 
@@ -66,8 +66,8 @@ def generate_prompt(text):
     return f"""Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n" +
                ### Instruction:\n\n{text}\n\n### Response:\n\n"""
 
-tokenizer = LlamaTokenizer.from_pretrained('BlueWhaleX/Chinese-Alpaca-COIG-49954-7B-HF')
-model = LlamaForCausalLM.from_pretrained('BlueWhaleX/Chinese-Alpaca-COIG-49954-7B-HF').half().cuda()
+tokenizer = LlamaTokenizer.from_pretrained('BlueWhaleX/bwx-7B-HF')
+model = LlamaForCausalLM.from_pretrained('BlueWhaleX/bwx-7B-HF').half().cuda()
 model.eval()
 
 text = '王国维说：“自周之衰，文王、周公势力之瓦解也，国民之智力成熟于内，政治之纷乱乘之于外，上无统一之制度，下迫于社会之要求，于是诸于九流各创其学说。” 他意在说明 A. 分封制的崩溃 B. 商鞅变法的作用 C. 兼并战争的后果 D. 百家争鸣的原因'
@@ -91,11 +91,6 @@ print("Response: ", response, '\n')
 
 ## Training Details
 
-### Training Data
-
-<!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
-
-BAAI/COIG-PC
 
 ### Training Procedure 
 
@@ -149,9 +144,9 @@ class TreeNode:
 
 <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
 ```
-@software{Chinese-Alpaca-COIG-49954-7B-HF,
-      title={An Enchanced Chinese Language Model based on the Chinese-Alpaca}, 
-      url={https://huggingface.co/BlueWhaleX/Chinese-Alpaca-COIG-49954-7B-HF},
+@software{bwx-7B-HF,
+      title={An Enchanced Chinese Language Model based on the Chinese-LLaMA-Alpaca}, 
+      url={https://huggingface.co/BlueWhaleX/bwx-7B-HF},
       year={2023}
 }
 ```