Eduardo Ottone
e331af2a48
- Enhanced tensor shape verification with detailed logging - Added failsafe tensor interpolation for shape mismatches - Comprehensive error handling with fallback options - Added reload_node.py script for development - Created TROUBLESHOOTING.md with step-by-step fixes - Detailed console output for debugging tensor issues - Prevents crashes by returning single object as fallback
43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Simple script to help reload the YOLOv8 detection node in ComfyUI
|
|
Run this script after making changes to the node code.
|
|
"""
|
|
|
|
import importlib
|
|
import sys
|
|
import os
|
|
|
|
def reload_node():
|
|
"""Attempt to reload the YOLOv8 detection node"""
|
|
try:
|
|
# Add current directory to Python path
|
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
if current_dir not in sys.path:
|
|
sys.path.insert(0, current_dir)
|
|
|
|
# Try to reload the module
|
|
module_name = 'yolov8_object_detection_node'
|
|
if module_name in sys.modules:
|
|
print(f"Reloading {module_name}...")
|
|
importlib.reload(sys.modules[module_name])
|
|
print("✅ Module reloaded successfully!")
|
|
else:
|
|
print(f"Module {module_name} not found in sys.modules")
|
|
print("Available modules:", [m for m in sys.modules.keys() if 'yolo' in m.lower()])
|
|
|
|
# Also try reloading the main init module
|
|
init_module = '__init__'
|
|
if init_module in sys.modules:
|
|
importlib.reload(sys.modules[init_module])
|
|
print("✅ Init module reloaded!")
|
|
|
|
except Exception as e:
|
|
print(f"❌ Error reloading module: {e}")
|
|
print("You may need to restart ComfyUI to load the updated node.")
|
|
|
|
if __name__ == "__main__":
|
|
print("🔄 Attempting to reload YOLOv8 detection node...")
|
|
reload_node()
|
|
print("\n📝 If this doesn't work, please restart ComfyUI completely.")
|