bitext
/

Mistral-7B-Travel

Text Generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions Community

malmarjeh commited on Aug 7

Commit

3237337

•

1 Parent(s): a5405e8

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -40,6 +40,7 @@ model = AutoModelForCausalLM.from_pretrained("bitext/Mistral-7B-Travel")
 tokenizer = AutoTokenizer.from_pretrained("bitext/Mistral-7B-Travel")
 messages = [
     {"role": "user", "content": "where to see flight tickets prices from chicago to madrid?"},
 ]

 tokenizer = AutoTokenizer.from_pretrained("bitext/Mistral-7B-Travel")
 messages = [
+    {"role": "system", "content": "You are an expert in customer support for Travel."},
     {"role": "user", "content": "where to see flight tickets prices from chicago to madrid?"},
 ]