xingxm commited on
Commit
b70838a
·
1 Parent(s): 40f9206

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, AnyPath): # mode 2
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