Commit
•
b633fcc
1
Parent(s):
f5b31f9
Update README.md
Browse files
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", "
|
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 = [
|