Update README.md
Browse files
README.md
CHANGED
@@ -46,7 +46,7 @@ tokenizer = AutoTokenizer.from_pretrained(finetuned_model, trust_remote_code=Tru
|
|
46 |
# Set pad_token_id to eos_token_id
|
47 |
model_pretrained.config.pad_token_id = tokenizer.eos_token_id
|
48 |
|
49 |
-
pipe = pipeline(task="text-generation", model=model_pretrained, tokenizer=tokenizer, max_length=
|
50 |
|
51 |
def build_prompt(question):
|
52 |
prompt = f"[INST]@Enlighten. [/INST] {question}"
|
|
|
46 |
# Set pad_token_id to eos_token_id
|
47 |
model_pretrained.config.pad_token_id = tokenizer.eos_token_id
|
48 |
|
49 |
+
pipe = pipeline(task="text-generation", model=model_pretrained, tokenizer=tokenizer, max_length=100)
|
50 |
|
51 |
def build_prompt(question):
|
52 |
prompt = f"[INST]@Enlighten. [/INST] {question}"
|