From 7c3e60be4c026c94f071a2d62fd4363fbe96d7de Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E3=81=B5=E3=81=81?= Date: Fri, 29 Nov 2024 13:18:35 +0900 Subject: [PATCH] fix sum providers bug MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: ふぁ --- rembg/sessions/sam.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/rembg/sessions/sam.py b/rembg/sessions/sam.py index 7a287cd..e6d8c4b 100644 --- a/rembg/sessions/sam.py +++ b/rembg/sessions/sam.py @@ -105,9 +105,10 @@ class SamSession(BaseSession): valid_providers = [] available_providers = ort.get_available_providers() - for provider in providers or []: - if provider in available_providers: - valid_providers.append(provider) + if providers: + for provider in providers or []: + if provider in available_providers: + valid_providers.append(provider) else: valid_providers.extend(available_providers)