Args' object has no attribute 'prior_loss_weight - [BUG]

[m-nameer]

Prerequisites

• I have read the documentation.
• I have checked other issues for similar problems.

Backend

Local

Interface Used

CLI

CLI Command

No response

UI Screenshots & Parameters

No response

Error Logs

ERROR | 2024-08-08 09:15:21 | autotrain.trainers.common:wrapper:120 - train has failed due to an exception: Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/autotrain/trainers/common.py", line 117, in wrapper
return func(*args, **kwargs)
File "/usr/local/lib/python3.10/dist-packages/autotrain/trainers/dreambooth/main.py", line 185, in train
main(_args)
File "/usr/local/lib/python3.10/dist-packages/autotrain/trainers/dreambooth/train.py", line 876, in main
loss = loss + args.prior_loss_weight * prior_loss
AttributeError: 'Args' object has no attribute 'prior_loss_weight'

ERROR | 2024-08-08 09:15:21 | autotrain.trainers.common:wrapper:121 - 'Args' object has no attribute 'prior_loss_weight'
Steps: 0% 0/500 [00:00<?, ?it/s]
INFO | 2024-08-08 09:15:24 | autotrain.cli.run_dreambooth:run:393 - Job ID: 8705

Additional Information

No response

[abhishekkrthakur]

please make sure you have the correct version of everything (including diffusers) from https://github.com/huggingface/autotrain-advanced/blob/main/requirements.txt

[github-actions]

This issue is stale because it has been open for 30 days with no activity.

[github-actions]

This issue was closed because it has been inactive for 20 days since being marked as stale.