add briaai/RMBG-2.0

https://github.com/Sanster/IOPaint/issues/598
2024-11-23 14:56:14 +08:00
parent ecbb73aaae
commit 175c597df8
5 changed files with 53 additions and 1 deletions
--- a/iopaint/schema.py
+++ b/iopaint/schema.py
@@ -145,6 +145,7 @@ class RealESRGANModel(Choices):

 class RemoveBGModel(Choices):
    briaai_rmbg_1_4 = "briaai/RMBG-1.4"
+    briaai_rmbg_2_0 = "briaai/RMBG-2.0"
    # models from https://github.com/danielgatis/rembg
    u2net = "u2net"
    u2netp = "u2netp"