Update README.md

README.md

@@ -32,10 +32,10 @@ encoded_input = tokenizer(prompt, return_tensors='pt')
 outputs = model(**encoded_input)
 
 token_embeddings = outputs.token_embeddings
-sentence_embedding = outputs.sentence_embedding
+sentence_embedding = outputs.sentence_embedding:
 
-print(f"token embeddings: {token_embeddings.shape}") # torch.Size([1, 7, 384])
-print(f"sentence_embedding: {sentence_embedding.shape}") # torch.Size([1, 384])
+print(f"token embeddings: {token_embeddings.shape}") # torch.Size([1, 7, 1024])
+print(f"sentence_embedding: {sentence_embedding.shape}") # torch.Size([1, 1024])
 
 ```