winglian
/

vicuna-13b-1_1-hf

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

winglian commited on Apr 22, 2023

Commit

307ad62

·

1 Parent(s): 423b228

fix end token

Files changed (1) hide show

instruct_pipeline.py +1 -1

instruct_pipeline.py CHANGED Viewed

@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
 INSTRUCTION_KEY = "USER:"
 RESPONSE_KEY = "ASSISTANT:"
-END_KEY = "</a>"
 INTRO_BLURB = (
     "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions."
 )

 INSTRUCTION_KEY = "USER:"
 RESPONSE_KEY = "ASSISTANT:"
+END_KEY = "</s>"
 INTRO_BLURB = (
     "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions."
 )