XavierXiao / Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
MIT License
7.6k stars 795 forks source link

name 'trainer' is not defined #165

Open b67294 opened 1 month ago

b67294 commented 1 month ago

while i was using python main.py --base configs/stable-diffusion/v1-finetune_unfrozen.yaml -t --actual_resume models/sd_ckpt/sd-v1-4-full-ema.ckpt --gpus 0,1, --data_root jellycat --reg_data_root outputs/regularization_images --class_word plush toy --resume logs/jellycat2024-09-30T21-21-26_v1-finetune_unfrozen to train my own model,it seems trainer can not be defined properly.

it says: During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "main.py", line 851, in if trainer.global_rank == 0: NameError: name 'trainer' is not defined

can anyone tell me how to slove this