Colab test on W4A1C4

Yes, version compatibility issue. You can try this.

Best,
Saif.