fix mask_dir

This commit is contained in:
Qing
2024-08-21 23:13:11 +08:00
parent 63131713cf
commit 89f02679d3
2 changed files with 3 additions and 0 deletions

View File

@@ -81,6 +81,8 @@ class FileManager:
@staticmethod @staticmethod
def _media_names(directory: Path) -> List[MediasResponse]: def _media_names(directory: Path) -> List[MediasResponse]:
if directory is None:
return []
names = sorted([it.name for it in glob_img(directory)]) names = sorted([it.name for it in glob_img(directory)])
res = [] res = []
for name in names: for name in names:

View File

@@ -44,6 +44,7 @@ default_configs = dict(
cpu_textencoder=False, cpu_textencoder=False,
device=Device.cuda, device=Device.cuda,
input=None, input=None,
mask_dir=None,
output_dir=None, output_dir=None,
quality=95, quality=95,
enable_interactive_seg=False, enable_interactive_seg=False,