search

JittorRepos / JDiffusion

JDiffusion is a diffusion model library for generating images or videos based on Diffusers and Jittor.
Apache License 2.0
250 stars 4 forks source link

KeyError when load_lora_weights in run_all.py #5

Open qcy22 opened 2 months ago

qcy22 commented 2 months ago

I added "conv" "conv1" and "conv2" to the target_modules .Now target_modules=["to_k", "to_q", "to_v", "to_out.0", "add_k_proj", "add_v_proj","conv1","conv2","conv"] .It trained well,but when pipe.load_lora_weights in run_all.py a key error was encountered.The following picture describes the error message

640c23941449e05191e6ec15742274c

jittor=1.3.9.8 peft= 0.10.0