Help for test model on Google Colab

Zheng-Chong / CatVTON

CatVTON is a simple and efficient virtual try-on diffusion model with 1) Lightweight Network (899.06M parameters totally), 2) Parameter-Efficient Training (49.57M parameters trainable) and 3) Simplified Inference (< 8G VRAM for 1024X768 resolution).

Other

951 stars 114 forks source link

Help for test model on Google Colab #41

Open arad2022 opened 2 months ago

arad2022 commented 2 months ago

Hi. Can you help me to run the model on Google Colab. I couldn't run it on Colab got out of a memory error. Also, app.py doesn't work on Colab.

Zheng-Chong commented 2 months ago

"Out of Memory" means that the GPU memory you are using is too small; try using a GPU with larger memory. The App will create a temporary webpage on localhost, which requires you to deploy in an environment with a browser or an environment that can forward ports. Colab may not support these.