Files
lebigdog f672cd8df9 🎯 Add comprehensive model management and installation tools
✨ Added:
- REQUIRED_MODELS.md - Complete guide for ControlNet model downloads
- download_models.py - Automatic model downloader with progress bars
- test_installation.py - Comprehensive installation testing suite
- models/README.md - Future model management documentation

🛠️ Updated:
- README.md - Clear installation instructions with model requirements
- Added troubleshooting section with common issues
- Added verification steps and diagnostic tools
- Highlighted critical ControlNet model requirement

🎯 Key Features:
- Automatic detection of ComfyUI installation path
- Progress bars for large model downloads (5.6GB total)
- Comprehensive system testing (Python, PyTorch, dependencies, models)
- Automatic issue detection and fixing attempts
- Clear error messages and solutions
- Verification of all 4 required ControlNet models

⚠️ Critical: Users now have clear guidance that 4 ControlNet models (~5.6GB) are REQUIRED for Kanibus to function properly.

Ready for production use with foolproof installation! 🚀
2025-08-03 21:13:05 -03:00
..

🧠 Kanibus Models Directory

📁 Model Storage Strategy

This directory is reserved for custom and specialized models that are not automatically downloaded by the standard libraries.


🎯 Current Status: EMPTY (By Design)

Kanibus v1.0.0 uses external models that are automatically downloaded:

🔄 Auto-Downloaded Models

  • MediaPipe Face Mesh - 468 landmarks + iris detection
  • Hugging Face Transformers - DPT depth estimation models
  • Ultralytics YOLO - Object detection and pose estimation
  • Segment Anything (SAM) - Advanced segmentation
  • PyTorch Vision Models - Various computer vision models

These models are cached in standard locations:

  • ~/.mediapipe/ - MediaPipe models
  • ~/.cache/huggingface/ - Transformer models
  • ~/.cache/torch/hub/ - PyTorch Hub models
  • ~/.ultralytics/ - YOLO models

🔮 Future Use Cases

This directory is prepared for:

📁 Custom Models (v1.1+)

models/
├── custom_eye_tracker_v2.pth      # Fine-tuned eye tracking
├── medical_gaze_analyzer.onnx     # Medical applications  
├── gaming_optimized_tracker.trt   # Gaming optimizations
├── emotion_micro_expressions.pth  # Advanced emotion AI
└── offline_models/                # Offline deployment models
    ├── face_mesh_offline.tflite
    ├── depth_estimation_lite.onnx
    └── pose_detection_mobile.trt

🎯 Specialized Models

  • Domain-specific - Medical, gaming, research variants
  • Performance-optimized - TensorRT, ONNX, TensorFlow Lite
  • Offline variants - For air-gapped environments
  • Fine-tuned models - Trained on specific datasets
  • Quantized models - For edge deployment

⚡ Performance Models

  • TensorRT engines - NVIDIA GPU optimization
  • ONNX models - Cross-platform inference
  • TensorFlow Lite - Mobile/edge deployment
  • CoreML models - Apple Silicon optimization

📋 Model Requirements

When adding custom models:

📝 Documentation

  • Add model description to this README
  • Include performance benchmarks
  • Document input/output specifications
  • Provide usage examples

🔧 Code Integration

  • Add model loading in relevant node files
  • Implement fallback to auto-downloaded models
  • Add model validation and error handling
  • Update configuration options

📊 Performance

  • Benchmark against baseline models
  • Test memory usage and speed
  • Validate accuracy metrics
  • Check multi-GPU compatibility

🚀 Adding Your Own Models

1. Place Model Files

# Copy your model files
cp your_model.pth models/
cp your_model.onnx models/

2. Update Node Code

# In your node file
model_path = os.path.join(os.path.dirname(__file__), "../models/your_model.pth")
if os.path.exists(model_path):
    custom_model = torch.load(model_path)
else:
    # Fallback to auto-downloaded model
    custom_model = load_default_model()

3. Update Documentation

  • Add to this README
  • Update node documentation
  • Include in CHANGELOG.md

📞 Support

For custom model integration:


This directory structure is prepared for future extensibility while keeping v1.0.0 simple and efficient.