Spaces:
Runtime error
Runtime error
Commit
·
930c104
1
Parent(s):
3bf2cbc
switch to hough
Browse files
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-
|
| 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 |
|