소스 검색

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)