| # DeOldify Modernization Roadmap | |
| This document tracks the remaining tasks to fully modernize the DeOldify repository, prioritized by user feedback. | |
| ## 🔴 Priority 1: Intel GPU Support (A-1) | |
| *Goal: Ensure robust support for Intel Arc and Data Center GPUs.* | |
| - [x] **Verify `device_id.py`** | |
| - Ensure `DeviceId` enum includes Intel/XPU options if necessary. | |
| - [x] **Verify Integration** | |
| - Check if `visualize.py` and other scripts correctly use the `_device.py` manager for Intel hardware. | |
| - Ensure fallback logic (Intel -> NVIDIA -> CPU) works as intended. | |
| - [ ] **Intel NPU Support** | |
| - Investigate and implement NPU detection (OpenVINO/IPEX). | |
| ## 🔴 Priority 2: NVIDIA GPU Modernization (A-2) | |
| *Goal: Leverage latest CUDA and PyTorch features while maintaining reasonable backward compatibility.* | |
| - [x] **Verify Optimization** | |
| - Ensure `torch.backends.cudnn.benchmark` and other optimizations are correctly applied for modern NVIDIA GPUs. | |
| - Double check `environment.yml` against latest PyTorch/CUDA matrix. | |
| - [x] **Legacy CUDA Support** | |
| - Document support for older drivers (last 5 years) where possible, or provide alternative environment files. | |
| ## 🟠 Priority 3: Colab Modernization | |
| *Goal: Fix broken notebooks and improve UX.* | |
| - [x] **Fix `requirements-colab.txt`** | |
| - Remove `fastai==1.0.60` (conflicts with new PyTorch). | |
| - Add `torch`, `torchvision`, and other modern dependencies. | |
| - [x] **Update Notebooks** | |
| - Remove `import fastai`. | |
| - Use `deoldify.fastai_compat`. | |
| - [x] **UI Improvements** | |
| - Drag-and-drop, Gallery, Comparison view. | |
| ## 🟡 Priority 4: Documentation & Testing | |
| *Goal: Comprehensive guides and stability.* | |
| - [x] **Detailed Documentation** | |
| - Create `docs/HARDWARE_GUIDE.md` (Benchmarks & Requirements). | |
| - Create `docs/DEPLOYMENT_GUIDE.md` (Local serving). | |
| - [x] **Testing** | |
| - Create `tests/` directory with unit tests for device selection and compatibility layer. | |
| - Add CI/CD workflow. | |
| ## 🔵 Future Work: Cloud Integration | |
| - [x] **Migrate Pretrained Weights** | |
| - Host models on GitHub Releases to ensure longevity. | |
| - Update all notebooks and docs to point to new URLs. | |
| - [ ] **Google Cloud Platform (Vertex AI)** | |
| - Deployment scripts and containers. | |