Do not send model to cuda internally (#154)

awslabs · Nov 20, 2023 · d054c8d · d054c8d
1 parent afbf206
commit d054c8d
Showing 1 changed file with 0 additions and 8 deletions.
diff --git a/fortuna/hallucination/base.py b/fortuna/hallucination/base.py
@@ -110,10 +110,6 @@ def fit(
         Dict
             The status returned by fitting the multicalibrator.
         """
-        self.generative_model.to(
-            torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        )
-
         (
             scores,
             embeddings,
@@ -177,10 +173,6 @@ def predict_proba(
         if self.multicalibrator is None:
             raise ValueError("`fit` must be called before this method.")
 
-        self.generative_model.to(
-            torch.device("cuda" if torch.cuda.is_available() else "cpu")
-        )
-
         (
             scores,
             embeddings,