The code [here](https://github.com/Atten4Vis/DemystifyLocalViT/blob/master/models/dwnet.py#L73) should be elif?