Backport PR #1451: fix prepare query data when model trained with cuda (

#1454) Co-authored-by: Adam Gayoso <[email protected]>
scverse · Mar 21, 2022 · 56d3ece · 56d3ece
1 parent a81645d
commit 56d3ece
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scvi/model/base/_archesmixin.py b/scvi/model/base/_archesmixin.py
@@ -195,7 +195,7 @@ def prepare_query_anndata(
         Query adata ready to use in `load_query_data` unless `return_reference_var_names`
         in which case a pd.Index of reference var names is returned.
         """
-        _, var_names, _ = _get_loaded_data(reference_model)
+        _, var_names, _ = _get_loaded_data(reference_model, device="cpu")
         var_names = pd.Index(var_names)
 
         if return_reference_var_names: