AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
fix

Co-authored-by: yiyixuxu <yixu@Yis-MacBook-Pro.local>
This commit is contained in:
YiYi Xu
2023-10-30 23:06:16 -10:00
committed by GitHub
parent ed00ead345
commit ce9484b139
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/kandinsky2_2/text_to_image/train_text_to_image_lora_prior.py
View File

@@ -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()