print plugin check_dep error
This commit is contained in:
@@ -95,8 +95,11 @@ class RemoveBG(BasePlugin):
|
|||||||
def check_dep(self):
|
def check_dep(self):
|
||||||
try:
|
try:
|
||||||
import rembg
|
import rembg
|
||||||
except ImportError:
|
except ImportError as e:
|
||||||
return "RemoveBG is not installed, please install it first. pip install -U rembg"
|
import traceback
|
||||||
|
|
||||||
|
error_msg = traceback.format_exc()
|
||||||
|
return f"Install rembg failed, Error details:\n{error_msg}"
|
||||||
|
|
||||||
def device_warning(self):
|
def device_warning(self):
|
||||||
if self.device == Device.cuda and self.model_name not in [
|
if self.device == Device.cuda and self.model_name not in [
|
||||||
|
|||||||
Reference in New Issue
Block a user