Update README.md

README.md

@@ -35,8 +35,8 @@ model = AutoModel.from_pretrained(checkpoint, trust_remote_code=True).to(device)
 
 inputs = tokenizer.encode("def print_hello_world():\tprint('Hello World!')", return_tensors="pt").to(device)
 embedding = model(inputs)[0]
-print(f'Dimension of the embedding: {embedding.size()[0]}, with norm={embedding.norm().item()}')
-print(embedding)
+print(f'Dimension of the embedding: {embedding[0].size()}')
+# Dimension of the embedding: torch.Size([13, 1024])
 ```
 
 ### BibTeX entry and citation info