Merge pull request #64 from x-tabdeveloping/ctm_hotfix

CTM hotfix
x-tabdeveloping · Sep 10, 2024 · aaa90a7 · aaa90a7
2 parents 21fd848 + 31ddd32
commit aaa90a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pyproject.toml b/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 <[email protected]>"]
 license = "MIT"

diff --git a/turftopic/models/ctm.py b/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: