From 1cd36cbe604b42c1de28cc29dca77bb7fd638eef Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Sat, 10 Feb 2024 13:19:43 +0800
Subject: [PATCH] briaai/RMBG-1.4 always run on cpu

---
 iopaint/plugins/briarmbg.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/iopaint/plugins/briarmbg.py b/iopaint/plugins/briarmbg.py
index baa7adf..880f530 100644
--- a/iopaint/plugins/briarmbg.py
+++ b/iopaint/plugins/briarmbg.py
@@ -490,9 +490,6 @@ def briarmbg_process(bgr_np_image, session, only_mask=False):
     im_tensor = torch.unsqueeze(im_tensor, 0)
     im_tensor = torch.divide(im_tensor, 255.0)
     im_tensor = normalize(im_tensor, [0.5, 0.5, 0.5], [1.0, 1.0, 1.0])
-    if torch.cuda.is_available():
-        im_tensor = im_tensor.cuda()
-
     # inference
     result = session(im_tensor)
     # post process