From ca6409fc7a8f946e2de7c0c94f0e8d36797425d1 Mon Sep 17 00:00:00 2001
From: Shojin Tam <99107633+shojint@users.noreply.github.com>
Date: Thu, 18 Jul 2024 11:13:03 +0800
Subject: [PATCH] Update lama_inpaint.py

Feature refinement does backward propagation for large images. @torch.no_grad() is removed to allow this.
---
 lama_inpaint.py | 1 -
 1 file changed, 1 deletion(-)

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,