philschmid HF staff commited on
Commit
b633fcc
1 Parent(s): f5b31f9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ _coming soon_
40
  from optimum.neuron import pipeline
41
 
42
  # Load pipeline from Hugging Face repository
43
- pipe = pipeline("text-generation", "meta-llama/Llama-2-7b-chat-hf")
44
 
45
  # We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
46
  messages = [
 
40
  from optimum.neuron import pipeline
41
 
42
  # Load pipeline from Hugging Face repository
43
+ pipe = pipeline("text-generation", "aws-neuron/Llama-2-7b-chat-hf-seqlen-2048-bs-4")
44
 
45
  # We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
46
  messages = [