# Intel GPU Setup Guide for DeOldify

This guide covers how to set up DeOldify with Intel GPUs (Arc A-Series, Data Center GPU Max Series) using Intel Extension for PyTorch (IPEX).

## Prerequisites

- **Intel GPU Driver**: Latest stable driver for your hardware.
- **Anaconda** or **Miniconda** installed.
- **Git** installed.
- **OS**: Windows 10/11 or Linux (Ubuntu 22.04 recommended).

## Installation Steps

1.  **Clone the Repository**
    ```bash
    git clone https://github.com/thookham/DeOldify.git
    cd DeOldify
    ```

2.  **Create Intel Conda Environment**
    We use a dedicated environment file that installs PyTorch with XPU support and Intel Extension for PyTorch.
    ```bash
    conda env create -f environment_intel.yml
    ```

3.  **Activate Environment**
    ```bash
    conda activate deoldify_intel
    ```

4.  **Download Weights**
    Download the pretrained weights and place them in the `models/` directory:
    - [ColorizeArtistic_gen.pth](https://github.com/thookham/DeOldify/releases/download/v2.0-models/ColorizeArtistic_gen.pth)
    - [ColorizeStable_gen.pth](https://github.com/thookham/DeOldify/releases/download/v2.0-models/ColorizeStable_gen.pth)
    - [ColorizeVideo_gen.pth](https://github.com/thookham/DeOldify/releases/download/v2.0-models/ColorizeVideo_gen.pth)

## Verification

Run the verification script to ensure your Intel GPU is detected:

```bash
python verify_refactor.py
```

If successful, you should see:
- `Imports successful`
- `Device detection passed`
- `Device Name: Intel(R) Arc(TM)...` (or similar)

## Troubleshooting

### "XPU not available"
- Ensure you have installed the correct drivers.
- Verify that `intel-extension-for-pytorch` is installed: `pip list | grep intel`
- On Windows, ensure you are using the correct oneAPI components if manually installed.

### Performance
- First run might be slower due to JIT compilation.
- Ensure you are using the `deoldify_intel` environment, not the standard one.