also removed from riemannian

facebookresearch · Dec 17, 2024 · be97497 · be97497
1 parent 1498339
commit be97497
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/flow_matching/solver/riemannian_ode_solver.py b/flow_matching/solver/riemannian_ode_solver.py
@@ -31,7 +31,6 @@ def __init__(self, manifold: Manifold, velocity_model: ModelWrapper):
         self.manifold = manifold
         self.velocity_model = velocity_model
 
-    @torch.no_grad()
     def sample(
         self,
         x_init: Tensor,