diff --git a/openvalidators/reward/diversity.py b/openvalidators/reward/diversity.py
index a7629a7..52eebd5 100644
--- a/openvalidators/reward/diversity.py
+++ b/openvalidators/reward/diversity.py
@@ -67,6 +67,7 @@ def get_embeddings( self, sentences: List[str] ) -> "torch.FloatTensor":
                 Embedding for the message.
         """
         # Tokenizing sentences
+
         encoded_input = self.tokenizer(
             sentences,
             padding=True,
@@ -87,6 +88,10 @@ def get_embeddings( self, sentences: List[str] ) -> "torch.FloatTensor":
 
     def get_rewards( self, prompt: str, completions: List[str], name: str ) -> torch.FloatTensor:
 
+        # Check if completions are empty, return 0 if so
+        if len(completions) == 0:
+            return torch.tensor([])
+        
         # Get embeddings for all completions.
         embeddings = self.get_embeddings( completions )