Merge pull request #64 from x-tabdeveloping/ctm_hotfix

CTM hotfix

Author: x-tabdeveloping

pyproject.toml

@@ -6,7 +6,7 @@ line-length=79
 
 [tool.poetry]
 name = "turftopic"
-version = "0.5.3"
+version = "0.5.4"
 description = "Topic modeling with contextual representations from sentence transformers."
 authors = ["Márton Kardos <power.up1163@gmail.com>"]
 license = "MIT"

turftopic/models/ctm.py

@@ -213,9 +213,7 @@ def fit(
             seed = self.random_state or random.randint(0, 10_000)
             torch.manual_seed(seed)
             pyro.set_rng_seed(seed)
-            device = torch.device(
-                "cuda:0" if torch.cuda.is_available() else "cpu"
-            )
+            device = torch.device("cpu")
             pyro.clear_param_store()
             contextualized_size = embeddings.shape[1]
             if self.combined: