Skip to content

Commit

Permalink
Remove debug logging
Browse files Browse the repository at this point in the history
  • Loading branch information
Lajcisvk committed Mar 21, 2024
1 parent b298cb6 commit b10e9b2
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 4 deletions.
3 changes: 0 additions & 3 deletions rembg/sessions/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,6 @@ def __init__(

self.providers.remove("TensorrtExecutionProvider")

print("Detected providers: %s" % _providers)
print("Selected providers: %s" % self.providers)

self.inner_session = ort.InferenceSession(
str(self.__class__.download_models(*args, **kwargs)),
providers=self.providers,
Expand Down
2 changes: 1 addition & 1 deletion rembg/sessions/sam.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ def __init__(self, model_name: str, sess_opts: ort.SessionOptions, *args, **kwar
"""
self.model_name = model_name
paths = self.__class__.download_models(*args, **kwargs)
print("SAM detected providers: %s" % ort.get_available_providers())

self.encoder = ort.InferenceSession(
str(paths[0]),
providers=ort.get_available_providers(),
Expand Down

0 comments on commit b10e9b2

Please sign in to comment.