Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,7 @@ model.resize_token_embeddings(len(tokenizer))
|
|
80 |
|
81 |
gen = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
|
82 |
|
83 |
-
instruction = "[INST] Write
|
84 |
res = gen(instruction, max_new_tokens=512, temperature=0.3, top_p=0.75, top_k=40, repetition_penalty=1.2)
|
85 |
print(res[0]['generated_text'])
|
86 |
```
|
|
|
80 |
|
81 |
gen = pipeline("text-generation", model=model, tokenizer=tokenizer, device=0)
|
82 |
|
83 |
+
instruction = "[INST] Write an email to say goodbye to me boss [\INST]"
|
84 |
res = gen(instruction, max_new_tokens=512, temperature=0.3, top_p=0.75, top_k=40, repetition_penalty=1.2)
|
85 |
print(res[0]['generated_text'])
|
86 |
```
|