(
self,
config=None,
rmbg_model_path: str = None,
)
| 244 | |
| 245 | |
| 246 | def __init__( |
| 247 | self, |
| 248 | config=None, |
| 249 | rmbg_model_path: str = None, |
| 250 | ): |
| 251 | super().__init__(config) |
| 252 | self._rmbg_model: Optional[RMBGModel] = None |
| 253 | self._rmbg_model_path = rmbg_model_path |
| 254 | |
| 255 | def load_rmbg_model(self): |
| 256 | """Load RMBG model.""" |