diff --git a/lama_inpaint.py b/lama_inpaint.py
index 20ebd9b1..fb9db1ba 100644
--- a/lama_inpaint.py
+++ b/lama_inpaint.py
@@ -24,7 +24,6 @@
 from utils import load_img_to_array, save_array_to_img
 
 
-@torch.no_grad()
 def inpaint_img_with_lama(
         img: np.ndarray,
         mask: np.ndarray,