Problem about the PyTorch vision?

[Wang-Xiaodong1899]

I have look for the issues but I can't find the same problem. So sorry to bother you.
GPU:

my python environment: pytorch=1.8.0&torchvision=0.9.0, cudatoolkit=11.3.1&cudnn =8.2.1. I have tried the rudalle=0.3.0 just following the readme.md, or 0.0.1rc5 by the RTX3090.ipynb, but I only got the following error!

So I wanna know if any problem in my environment? Waiting for your reply!

[shonenkov]

@Wang-xd1899 could you try to set mlp_activation="gelu"?

get_rudalle_model('Malevich', pretrained=True, fp16=True, device=device, mlp_activation='gelu')

[Wang-Xiaodong1899]

Surprisingly, after set mlp_activation="gelu", it works! Thank you very much!!!