From c4785e96d23cf64f76ad866c55f20590a7f8b42a Mon Sep 17 00:00:00 2001 From: Ikko Eltociear Ashimine Date: Fri, 22 Mar 2024 01:10:03 +0900 Subject: [PATCH] Update modeling_opt.py consciours -> conscious --- model/modeling_opt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model/modeling_opt.py b/model/modeling_opt.py index 5365d67..dbefca5 100644 --- a/model/modeling_opt.py +++ b/model/modeling_opt.py @@ -920,13 +920,13 @@ def forward( >>> model = OPTForCausalLM.from_pretrained("facebook/opt-350m") >>> tokenizer = AutoTokenizer.from_pretrained("facebook/opt-350m") - >>> prompt = "Hey, are you consciours? Can you talk to me?" + >>> prompt = "Hey, are you conscious? Can you talk to me?" >>> inputs = tokenizer(prompt, return_tensors="pt") >>> # Generate >>> generate_ids = model.generate(inputs.input_ids, max_length=30) >>> tokenizer.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0] - "Hey, are you consciours? Can you talk to me?\nI'm not consciours, but I can talk to you." + "Hey, are you conscious? Can you talk to me?\nI'm not conscious, but I can talk to you." ```""" output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions