kaupane's picture
Update models/DiT.py
a7df152 verified