Update scvi/external/sysvi/_model.py

Hrovatin · Mar 15, 2024 · 0f7bd06 · 0f7bd06
1 parent e4c1ef9
commit 0f7bd06
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scvi/external/sysvi/_model.py b/scvi/external/sysvi/_model.py
@@ -110,7 +110,7 @@ def __init__(
 
         logger.info("The model has been initialized")
 
-    @torch.no_grad()
+    @torch.inference_mode()
     def get_latent_representation(
         self,
         adata: AnnData,