AttributeError: Can’t get attribute ‘RivaGAN’ on <module '__main__' from '/data/home/username/miniconda3/envs/video/bin/pytest'
I run the code in pytest and error at import a pt file as name “rivagan_32bit_model.pt”.It’s also normally running at main like