Update README.md

README.md

@@ -39,7 +39,7 @@ 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[0].size()}')
-# Dimension of the embedding: torch.Size([13, 1024])
+# Dimension of the embedding: torch.Size([14, 1024])
 ```
 
 ### BibTeX entry and citation info