浏览代码

fix sum providers bug

Signed-off-by: ふぁ <[email protected]>
ふぁ 9 月之前
父节点
当前提交
7c3e60be4c
共有 1 个文件被更改,包括 4 次插入3 次删除
  1. 4 3
      rembg/sessions/sam.py

+ 4 - 3
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)