Update timm universal (support transformer-style model)

Update timm universal (support transformer-style model) #621