diff --git a/src/laion_clap/hook.py b/src/laion_clap/hook.py
index fac512f..a97b6fa 100644
--- a/src/laion_clap/hook.py
+++ b/src/laion_clap/hook.py
@@ -69,7 +69,7 @@ def tokenizer(self, text):
             max_length=77,
             return_tensors="pt",
         )
-        return {k: v.squeeze(0) for k, v in result.items()}
+        return result
 
     def load_ckpt(self, ckpt = None, model_id = -1, verbose = True):
         """Load the pretrained checkpoint of CLAP model