search

IrisRainbowNeko / HCP-Diffusion

A universal Stable-Diffusion toolbox
Apache License 2.0
899 stars 75 forks source link

readme file not desc of train, I hope get some help! And I meet err. #49

Closed yudongjian closed 10 months ago

yudongjian commented 10 months ago

Readme file not desc of train, I hope get some help! I run accelerate launch -m hcpdiff.train_ac_single --cfg cfgs/train/examples/DreamArtist.yaml

I meet a err. I modify this "pretrained_model_name_or_path" config in DreamArtist.yaml Is there anything else that needs to be modified?
or give a yaml is direct run ! thanks! thanks! thanks!

Traceback (most recent call last):
  File "/home/image1325_user/anaconda3/envs/yudongjian_23_ldm/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/image1325_user/anaconda3/envs/yudongjian_23_ldm/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/home/image1325_user/ssd_disk1/yudongjian_23/HCP-Diffusion-main/hcpdiff/train_ac_single.py", line 44, in <module>
    trainer = TrainerSingleCard(conf)
  File "/home/image1325_user/ssd_disk1/yudongjian_23/HCP-Diffusion-main/hcpdiff/train_ac.py", line 78, in __init__
    self.build_optimizer_scheduler()
  File "/home/image1325_user/ssd_disk1/yudongjian_23/HCP-Diffusion-main/hcpdiff/train_ac.py", line 366, in build_optimizer_scheduler
    parameters, parameters_pt = self.get_param_group_train()
  File "/home/image1325_user/ssd_disk1/yudongjian_23/HCP-Diffusion-main/hcpdiff/train_ac.py", line 356, in get_param_group_train
    word_emb = self.ex_words_emb[v.name]
KeyError: 'pt-catgirl1'
IrisRainbowNeko commented 10 months ago

You need to create a custom word firstly. You can follow this