瀏覽代碼

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 = []
         valid_providers = []
         available_providers = ort.get_available_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:
         else:
             valid_providers.extend(available_providers)
             valid_providers.extend(available_providers)