File size: 2,261 Bytes
e9f9fd3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
# 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.
|