소스 검색

fix linters

Daniel Gatis 1 년 전
부모
커밋
6f1dd31ee6
1개의 변경된 파일3개의 추가작업 그리고 4개의 파일을 삭제
  1. 3 4
      rembg/sessions/sam.py

+ 3 - 4
rembg/sessions/sam.py

@@ -105,10 +105,9 @@ class SamSession(BaseSession):
         self.providers = []
 
         _providers = ort.get_available_providers()
-        if providers:
-            for provider in providers:
-                if provider in _providers:
-                    self.providers.append(provider)
+        for provider in kwargs.get("providers", []):
+            if provider in _providers:
+                self.providers.append(provider)
         else:
             self.providers.extend(_providers)