How much memory is required to run code test_tryon_image and can it be used in Colab? I am running the example in Google Colab, but it shows that there is not enough RAM memory. Is there a way to optimize it or is there a Colab example available
Closed this issue · 1 comments
libai-lab commented
How much memory is required to run code test_tryon_image and can it be used in Colab? I am running the example in Google Colab, but it shows that there is not enough RAM memory. Is there a way to optimize it or is there a Colab example available
danbochman commented
Just the model weights for the (cascaded) model are around 11GB so it makes sense that with the standard Colab machine (12 GB RAM) it won't be enough.
You should have around 16GB RAM and 24GB GPU memory to run this model.