Unverified Commit ce9484b1 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix a mistake in text2image training script for kandinsky2.2 (#5244)



fix
Co-authored-by: default avataryiyixuxu <yixu@Yis-MacBook-Pro.local>
parent ed00ead3
......@@ -682,7 +682,7 @@ def main():
# Backpropagate
accelerator.backward(loss)
if accelerator.sync_gradients:
accelerator.clip_grad_norm_(prior.parameters(), args.max_grad_norm)
accelerator.clip_grad_norm_(lora_layers.parameters(), args.max_grad_norm)
optimizer.step()
lr_scheduler.step()
optimizer.zero_grad()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment