From bc54f159b6e8b2da8cb25173bf8b39e74c781cf5 Mon Sep 17 00:00:00 2001 From: logtd Date: Fri, 23 Feb 2024 16:05:55 -0700 Subject: [PATCH] corrected models path --- models/place_models_here | 0 nodes/yolo_tracker_node.py | 7 ++----- 2 files changed, 2 insertions(+), 5 deletions(-) delete mode 100644 models/place_models_here diff --git a/models/place_models_here b/models/place_models_here deleted file mode 100644 index e69de29..0000000 diff --git a/nodes/yolo_tracker_node.py b/nodes/yolo_tracker_node.py index cb60709..827eb85 100644 --- a/nodes/yolo_tracker_node.py +++ b/nodes/yolo_tracker_node.py @@ -6,13 +6,10 @@ from PIL import Image import supervision as sv from ultralytics import YOLO -folder_paths.folder_names_and_paths["yolov8"] = ([os.path.join( - folder_paths.models_dir, "yolov8")], folder_paths.supported_pt_extensions) - def get_model_dir(): - root_path = folder_paths.folder_names_and_paths['custom_nodes'][0][0] - path = os.path.join(root_path, 'ComfyUI-TrackingNodes', 'models') + root_path = folder_paths.models_dir + path = os.path.join(root_path, 'yolov8') return path