From be974977d060030a272c402da6631fa04654ea7c Mon Sep 17 00:00:00 2001
From: Marton Havasi <marton@meta.com>
Date: Tue, 17 Dec 2024 10:49:05 +0000
Subject: [PATCH] also removed from riemannian

---
 flow_matching/solver/riemannian_ode_solver.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/flow_matching/solver/riemannian_ode_solver.py b/flow_matching/solver/riemannian_ode_solver.py
index 6eb3e5e..6f5ebcb 100644
--- 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,