Update rembg dependency and add BiRefNet models in the list of available RemBG models #47
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Hi,
The goal of this PR is to add BiRefNet models to the list of available remove background models. It follows discussion here and merged PR in rembg here.
The following modifications have been made:
rembg
even if a previous one was already installed.isnet-general-use
entry from the list of classic models.providers
when callingrembg
from UI or API with onlyCUDAExecutionProvider
andCPUExecutionProvider
as by default all available providers may be used, for instanceTensorrtExecutionProvider
orAzureExecutionProvider
, even if no external dependency is installed with this extension, leading to an error message in the console. Internally,rembg
will removeCUDAExecutionProvider
from its list of providers if it's not an available provider (cf. here).Please let me know if I need to make any changes.