fix(pipline): typing error
Browse files
svgdreamer/pipelines/SVGDreamer_pipeline.py
CHANGED
@@ -2,6 +2,7 @@
|
|
2 |
# Copyright (c) XiMing Xing. All rights reserved.
|
3 |
# Author: XiMing Xing
|
4 |
# Description:
|
|
|
5 |
import pathlib
|
6 |
from PIL import Image
|
7 |
from typing import AnyStr, Union, Tuple, List
|
@@ -454,7 +455,7 @@ class SVGDreamerPipeline(ModelState):
|
|
454 |
# create svg renderer
|
455 |
if isinstance(init_svg_path, List): # mode 3
|
456 |
renderers = [self.load_renderer(init_path) for init_path in init_svg_path]
|
457 |
-
elif isinstance(init_svg_path,
|
458 |
renderers = [self.load_renderer(init_svg_path) for _ in range(n_particle)]
|
459 |
init_image = [init_image] * n_particle
|
460 |
else: # mode 1
|
|
|
2 |
# Copyright (c) XiMing Xing. All rights reserved.
|
3 |
# Author: XiMing Xing
|
4 |
# Description:
|
5 |
+
import os
|
6 |
import pathlib
|
7 |
from PIL import Image
|
8 |
from typing import AnyStr, Union, Tuple, List
|
|
|
455 |
# create svg renderer
|
456 |
if isinstance(init_svg_path, List): # mode 3
|
457 |
renderers = [self.load_renderer(init_path) for init_path in init_svg_path]
|
458 |
+
elif isinstance(init_svg_path, (str, pathlib.Path, os.PathLike)): # mode 2
|
459 |
renderers = [self.load_renderer(init_svg_path) for _ in range(n_particle)]
|
460 |
init_image = [init_image] * n_particle
|
461 |
else: # mode 1
|