RamAnanth1 commited on
Commit
930c104
·
1 Parent(s): 3bf2cbc

switch to hough

Browse files
Files changed (1) hide show
  1. model.py +2 -3
model.py CHANGED
@@ -18,14 +18,13 @@ repo_dir = pathlib.Path(__file__).parent
18
  submodule_dir = repo_dir / 'ControlNet'
19
  sys.path.append(submodule_dir.as_posix())
20
 
21
-
22
- from annotator.midas import apply_midas
23
  from annotator.uniformer import apply_uniformer
24
  from annotator.util import HWC3, resize_image
25
 
26
  CONTROLNET_MODEL_IDS = {
27
 
28
- 'hough': 'lllyasviel/sd-controlnet-hough',
29
 
30
  }
31
 
 
18
  submodule_dir = repo_dir / 'ControlNet'
19
  sys.path.append(submodule_dir.as_posix())
20
 
21
+ from annotator.mlsd import apply_mlsd
 
22
  from annotator.uniformer import apply_uniformer
23
  from annotator.util import HWC3, resize_image
24
 
25
  CONTROLNET_MODEL_IDS = {
26
 
27
+ 'hough': 'lllyasviel/sd-controlnet-mlsd',
28
 
29
  }
30