>>> batch_size = len(prompt)
```  
Tokenize the text and generate the embeddings from the prompt:  
```py
>>> text_input = tokenizer(
...     prompt, padding="max_length", max_length=tokenizer.model_max_length, truncation=True, return_tensors="pt"
... )