Update text_generation.py

text_generation.py

@@ -55,19 +55,24 @@ class TextGenerationPipeline(Pipeline):
             truncation=True,
             max_length=english_tokens_max_length,
         ).input_ids
-        bio_tokens = self.bio_tokenizer(
-            dna_sequences,
-            return_tensors="pt",
-            padding="max_length",
-            max_length=bio_tokens_max_length,
-            truncation=True,
-        ).input_ids.unsqueeze(0)
+        if len(dna_sequences) == 0:
+            bio_tokens = None
+        else:
+            bio_tokens = self.bio_tokenizer(
+                dna_sequences,
+                return_tensors="pt",
+                padding="max_length",
+                max_length=bio_tokens_max_length,
+                truncation=True,
+            ).input_ids.unsqueeze(0)
 
         return {"english_tokens": english_tokens, "bio_tokens": bio_tokens}
 
     def _forward(self, model_inputs: dict, max_num_tokens_to_decode: int = 50) -> dict:
         english_tokens = model_inputs["english_tokens"].clone()
-        bio_tokens = model_inputs["bio_tokens"].clone()
+        bio_tokens = model_inputs["bio_tokens"]
+        if bio_tokens is not None:
+            bio_tokens = bio_tokens.clone()
         projected_bio_embeddings = None
 
         actual_num_steps = 0