Update README.md

README.md

@@ -42,7 +42,7 @@ To use Llamba with PyTorch:
 from transformers import AutoTokenizer
 from cartesia_pytorch.Llamba.llamba import LlambaLMHeadModel
 
-model = LlambaLMHeadModel.from_pretrained("cartesia-ai/Llamba-8B", strict=True).to('cuda')
+model = LlambaLMHeadModel.from_pretrained("cartesia-ai/Llamba-8B-untied", strict=True).to('cuda')
 tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B")
 input_ids = tokenizer("Hello, my name is", return_tensors="pt").input_ids
 input_ids = input_ids.to('cuda')