IGEV++ on Colab (Stereo Matching)

I need help running IGEV++ on Colab.

I tried IGEV++_01.ipynb but get errors - see in notebook. And I want to calc my own images.

Tasks:
1) Get demo run on Colab
2) Get it run with my images (attached: test-1_A1.png amd test-2_A1.png)
3) Load middlebury model and calculate demo images as in 1) and my images as in 2). Download for middlebury model:
https://drive.google.com/drive/folders/19WA5WeoSQGFxxyzq7ZhJFeptz4zwBNuq and there use middlebury.pth

General Requirements:
- The Jupyter Notebooks must run successfully on Google Colab
- The notebooks must be provided as part of the deliverables and must execute without errors on my side.
- You must use your own Colab account for development and testing.
- Only commercially allowed packages can be used, e.g. MIT licence
- Include thorough documentation within the code to explain each step.
- Payment will only be made upon successful completion of all project tasks (see requirements above).
- All communication and transactions must be conducted exclusively through Freelancer.