Merge pull request #130 from x-tabdeveloping/bfloat_fix

Bfloat fix

Author: x-tabdeveloping

pyproject.toml

@@ -9,7 +9,7 @@ profile = "black"
 
 [project]
 name = "turftopic"
-version = "0.25.1"
+version = "0.25.2"
 description = "Topic modeling with contextual representations from sentence transformers."
 authors = [
    { name = "Márton Kardos <power.up1163@gmail.com>", email = "martonkardos@cas.au.dk" }

turftopic/late.py

@@ -54,7 +54,10 @@ def _encode_tokens(
         """
         self.has_used_token_level = True
         token_embeddings = self.encode(
-            texts, output_value="token_embeddings", batch_size=batch_size
+            texts,
+            output_value="token_embeddings",
+            batch_size=batch_size,
+            show_progress_bar=show_progress_bar,
         )
         offsets = self.tokenizer(
             texts, return_offsets_mapping=True, verbose=False
@@ -63,7 +66,7 @@ def _encode_tokens(
             offs[: len(embs)] for offs, embs in zip(offsets, token_embeddings)
         ]
         token_embeddings = [
-            embs.numpy(force=True)
+            embs.float().numpy(force=True)
             for embs in token_embeddings
             if torch.is_tensor(embs)
         ]