primerz commited on
Commit
dc93517
·
verified ·
1 Parent(s): 3480bce

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +3 -2
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 ZoeDetector
 
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 = ZoeDetector.from_pretrained(
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)