fix: return numpy tensors from tokenizer

ml-gde · Sep 13, 2024 · 25c2bf1 · 25c2bf1
1 parent 76c05b2
commit 25c2bf1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jflux/conditioner.py b/jflux/conditioner.py
@@ -39,7 +39,7 @@ def forward(self, text: list[str]) -> Array:
             return_length=False,
             return_overflowing_tokens=False,
             padding="max_length",
-            return_tensors="pt",
+            return_tensors="np",
         )
 
         outputs = self.hf_module(