vicky4s4s
/

mistral-7b-instruct-v3

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

vicky4s4s commited on Jul 30, 2024

Commit

86a5cd1

·

verified ·

1 Parent(s): aafc0ce

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -14,7 +14,7 @@ license: apache-2.0
 import torch
 from transformers import pipeline
-pipe = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0", torch_dtype=torch.bfloat16, device_map="auto")
 # We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
 messages = [

 import torch
 from transformers import pipeline
+pipe = pipeline("text-generation", model="vicky4s4s/mistral-7b-instruct-v3", torch_dtype=torch.bfloat16, device_map="auto")
 # We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
 messages = [