Skip to content

GitLab

Commit 807226db authored by April Hu's avatar April Hu
Browse files

Fix typo

parent b7e71b41
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
comfyui/nodes.py
View file @ 807226db
...@@ -361,7 +361,7 @@ class DepthPreprocesser: ...@@ -361,7 +361,7 @@ class DepthPreprocesser:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
model_paths = ["LiheYoung/depth-anything-large-hf"] model_paths = ["LiheYoung/depth-anything-large-hf"]
prefix = "models/style_models" prefix = "models/checkpoints"
local_folders = os.listdir(prefix) local_folders = os.listdir(prefix)
local_folders = sorted( local_folders = sorted(
[ [
...@@ -388,7 +388,7 @@ class DepthPreprocesser: ...@@ -388,7 +388,7 @@ class DepthPreprocesser:
TITLE = "Flux.1 Depth Preprocessor" TITLE = "Flux.1 Depth Preprocessor"
def depth_preprocess(self, image, model_path): def depth_preprocess(self, image, model_path):
prefix = "models/style_models" prefix = "models/checkpoints"
if os.path.exists(os.path.join(prefix, model_path)): if os.path.exists(os.path.join(prefix, model_path)):
model_path = os.path.join(prefix, model_path) model_path = os.path.join(prefix, model_path)
processor = DepthPreprocessor.from_pretrained(model_path) processor = DepthPreprocessor.from_pretrained(model_path)
...@@ -402,5 +402,5 @@ NODE_CLASS_MAPPINGS = { ...@@ -402,5 +402,5 @@ NODE_CLASS_MAPPINGS = {
"SVDQuantFluxDiTLoader": SVDQuantFluxDiTLoader, "SVDQuantFluxDiTLoader": SVDQuantFluxDiTLoader,
"SVDQuantTextEncoderLoader": SVDQuantTextEncoderLoader, "SVDQuantTextEncoderLoader": SVDQuantTextEncoderLoader,
"SVDQuantLoRALoader": SVDQuantLoraLoader, "SVDQuantLoRALoader": SVDQuantLoraLoader,
"DepthPreprocesser": DepthPreprocesser, "DepthPreprocesser": DepthPreprocesser
} }
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment