Daniel Gatis hace 1 año
padre
commit
6f1dd31ee6
Se han modificado 1 ficheros con 3 adiciones y 4 borrados
  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)