[BUG] Error in train.py

[erlebach]

In train.py, I see the lines:

    model = model().to(accelerator.device) 

This did not work on my mac, because model() requires an argument since it calls the forward method.
The following line works correctly on the mac:

    model = model.to(accelerator.device)

I do not understand why the code worked at all. Thanks.

Upvote & Fund

• We're using Polar.sh so you can upvote and help fund this issue.
• We receive the funding once the issue is completed & confirmed by you.
• Thank you in advance for helping prioritize & fund our backlog.