rinongal / textual_inversion

MIT License
2.87k stars 278 forks source link

When I run it, I get the following error, #164

Open pcxdi opened 6 months ago

pcxdi commented 6 months ago

When I run it, python main.py --base configs/latent-diffusion/txt2img-1p4B-finetune.yaml -t --actual_resume models/ldm/text2img-large/model.ckpt -n img --gpus 0, --data_root I get the following error:image/real47 --init_word dog Global seed set to 23 Running on GPUs 0, Loading model from models/ldm/text2img-large/model.ckpt LatentDiffusion: Running in eps-prediction mode Killed I am running under 4060 8GB running memory, is this configuration OK

rinongal commented 6 months ago

Killed implies you're running out of memory. Can you try reducing the batch size and increasing the number of accumulation steps instead?