Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ model_type: yi
|
|
11 |
|
12 |
<!-- Provide a quick summary of what the model is/does. -->
|
13 |
|
14 |
-
dragon-yi-6b-0.1 part of the dRAGon ("Delivering RAG On
|
15 |
|
16 |
DRAGON models are fine-tuned with high-quality custom instruct datasets, designed for production quality use in RAG scenarios.
|
17 |
|
@@ -38,7 +38,7 @@ For test run results (and good indicator of target use cases), please see the fi
|
|
38 |
- **Developed by:** llmware
|
39 |
- **Model type:** Yi
|
40 |
- **Language(s) (NLP):** English
|
41 |
-
- **License:**
|
42 |
- **Finetuned from model:** Yi-6B
|
43 |
|
44 |
## Uses
|
@@ -84,7 +84,7 @@ tokenizer = AutoTokenizer.from_pretrained("dragon-yi-6b-0.1")
|
|
84 |
model = AutoModelForCausalLM.from_pretrained("dragon-yi-6b-0.1")
|
85 |
|
86 |
|
87 |
-
The
|
88 |
|
89 |
full_prompt = "\<human>\: " + my_prompt + "\n" + "\<bot>\:"
|
90 |
|
|
|
11 |
|
12 |
<!-- Provide a quick summary of what the model is/does. -->
|
13 |
|
14 |
+
dragon-yi-6b-0.1 part of the dRAGon ("Delivering RAG On ...") model series, RAG-instruct trained on top of a Yi-6B base model.
|
15 |
|
16 |
DRAGON models are fine-tuned with high-quality custom instruct datasets, designed for production quality use in RAG scenarios.
|
17 |
|
|
|
38 |
- **Developed by:** llmware
|
39 |
- **Model type:** Yi
|
40 |
- **Language(s) (NLP):** English
|
41 |
+
- **License:** Yi License (Link)<https://huggingface.co/01-ai/Yi-6B/blob/main/LICENSE>
|
42 |
- **Finetuned from model:** Yi-6B
|
43 |
|
44 |
## Uses
|
|
|
84 |
model = AutoModelForCausalLM.from_pretrained("dragon-yi-6b-0.1")
|
85 |
|
86 |
|
87 |
+
The DRAGON model was fine-tuned with a simple "\<human> and \<bot> wrapper", so to get the best results, wrap inference entries as:
|
88 |
|
89 |
full_prompt = "\<human>\: " + my_prompt + "\n" + "\<bot>\:"
|
90 |
|