AssertionError: torch version must be <= 1.10 in order to use OpenAI discrete vae

[mvaghei78]

Hello everyone,
I want to use OpenAI's Pretrained VAE model (I mean OpenAIDiscreteVAE() class ) for training DALLE, but I got this error:

AssertionError: torch version must be <= 1.10 in order to use OpenAI discrete vae

I tried to install torch versions less than 1.10 on google Colab:
!pip install torch==1.10.0

but I got this error:

ERROR: Could not find a version that satisfies the requirement torch==1.10.0 (from versions: 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1)
ERROR: No matching distribution found for torch==1.10.0

so there is no available version less than 1.11.0 for torch library in Colab. Is there any way to solve this problem?
I would appreciate your help. thanks.