Spaces:
Runtime error
Runtime error
Update models.py
Browse files
models.py
CHANGED
|
@@ -13,7 +13,8 @@ from diffusers import (
|
|
| 13 |
from diffusers.models.attention_processor import AttnProcessor2_0
|
| 14 |
from transformers import CLIPVisionModelWithProjection
|
| 15 |
from insightface.app import FaceAnalysis
|
| 16 |
-
from controlnet_aux import
|
|
|
|
| 17 |
from huggingface_hub import hf_hub_download
|
| 18 |
from compel import Compel, ReturnedEmbeddingsType
|
| 19 |
|
|
@@ -84,7 +85,7 @@ def load_depth_detector():
|
|
| 84 |
"""Load Zoe Depth detector."""
|
| 85 |
print("Loading Zoe Depth detector...")
|
| 86 |
try:
|
| 87 |
-
zoe_depth =
|
| 88 |
"lllyasviel/Annotators"
|
| 89 |
)
|
| 90 |
zoe_depth.to(device)
|
|
|
|
| 13 |
from diffusers.models.attention_processor import AttnProcessor2_0
|
| 14 |
from transformers import CLIPVisionModelWithProjection
|
| 15 |
from insightface.app import FaceAnalysis
|
| 16 |
+
from controlnet_aux import LeresDetector
|
| 17 |
+
from controlnet_aux.processor import Processor
|
| 18 |
from huggingface_hub import hf_hub_download
|
| 19 |
from compel import Compel, ReturnedEmbeddingsType
|
| 20 |
|
|
|
|
| 85 |
"""Load Zoe Depth detector."""
|
| 86 |
print("Loading Zoe Depth detector...")
|
| 87 |
try:
|
| 88 |
+
zoe_depth = LeresDetector.from_pretrained(
|
| 89 |
"lllyasviel/Annotators"
|
| 90 |
)
|
| 91 |
zoe_depth.to(device)
|