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)