Update README.md
Browse files
README.md
CHANGED
@@ -78,14 +78,14 @@ Any model can provide inaccurate or incomplete information, and should be used i
|
|
78 |
|
79 |
The fastest way to get started with BLING is through direct import in transformers:
|
80 |
|
81 |
-
from transformers import AutoTokenizer, AutoModelForCausalLM
|
82 |
-
tokenizer = AutoTokenizer.from_pretrained("dragon-yi-6b-0.1")
|
83 |
-
model = AutoModelForCausalLM.from_pretrained("dragon-yi-6b-0.1")
|
84 |
|
85 |
|
86 |
The DRAGON model was fine-tuned with a simple "\<human> and \<bot> wrapper", so to get the best results, wrap inference entries as:
|
87 |
|
88 |
-
full_prompt = "\<human>\: " + my_prompt + "\n" + "\<bot>\:"
|
89 |
|
90 |
The BLING model was fine-tuned with closed-context samples, which assume generally that the prompt consists of two sub-parts:
|
91 |
|
@@ -94,7 +94,7 @@ The BLING model was fine-tuned with closed-context samples, which assume general
|
|
94 |
|
95 |
To get the best results, package "my_prompt" as follows:
|
96 |
|
97 |
-
my_prompt = {{text_passage}} + "\n" + {{question/instruction}}
|
98 |
|
99 |
|
100 |
|
|
|
78 |
|
79 |
The fastest way to get started with BLING is through direct import in transformers:
|
80 |
|
81 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
82 |
+
tokenizer = AutoTokenizer.from_pretrained("dragon-yi-6b-0.1")
|
83 |
+
model = AutoModelForCausalLM.from_pretrained("dragon-yi-6b-0.1")
|
84 |
|
85 |
|
86 |
The DRAGON model was fine-tuned with a simple "\<human> and \<bot> wrapper", so to get the best results, wrap inference entries as:
|
87 |
|
88 |
+
full_prompt = "\<human>\: " + my_prompt + "\n" + "\<bot>\:"
|
89 |
|
90 |
The BLING model was fine-tuned with closed-context samples, which assume generally that the prompt consists of two sub-parts:
|
91 |
|
|
|
94 |
|
95 |
To get the best results, package "my_prompt" as follows:
|
96 |
|
97 |
+
my_prompt = {{text_passage}} + "\n" + {{question/instruction}}
|
98 |
|
99 |
|
100 |
|