Update README.md

pull/520/head
Alexander Borzunov 8 months ago committed by GitHub
parent b3c8b62ebe
commit d7335cef79
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -15,7 +15,7 @@ from transformers import AutoTokenizer
from petals import AutoDistributedModelForCausalLM
# Choose any model available at https://health.petals.dev
model_name = "petals-team/StableBeluga2" # This one is a fine-tuned variant of Llama 2 (70B)
model_name = "petals-team/StableBeluga2" # This one is fine-tuned Llama 2 (70B)
# Connect to a distributed network hosting model layers
tokenizer = AutoTokenizer.from_pretrained(model_name)

Loading…
Cancel
Save