AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-18 08:45:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix sum providers bug

Browse Source 
Signed-off-by: ふぁ <yuki@yuki0311.com>
This commit is contained in:
ふぁ 2024-11-29 13:18:35 +09:00
parent e740a9681e
commit 7c3e60be4c
No known key found for this signature in database
GPG Key ID: 83A8A5E74872A8AA
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
rembg/sessions/sam.py
View File

@ -105,9 +105,10 @@ class SamSession(BaseSession):
valid_providers = [] valid_providers = []
available_providers = ort.get_available_providers() available_providers = ort.get_available_providers()
for provider in providers or []: if providers:
if provider in available_providers: for provider in providers or []:
valid_providers.append(provider) if provider in available_providers:
valid_providers.append(provider)
else: else:
valid_providers.extend(available_providers) valid_providers.extend(available_providers)