From 92113e3a7c9cfda2bc2f6cc0fa5b1234505f145d Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Wed, 11 Jan 2023 21:54:10 +0100
Subject: TI: Use grad clipping from LoRA #104

---
 training/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'training')

diff --git a/training/common.py b/training/common.py
index 67c2ab6..0b2ae44 100644
--- a/training/common.py
+++ b/training/common.py
@@ -58,7 +58,7 @@ def generate_class_images(
             torch.cuda.empty_cache()
 
 
-def run_model(
+def loss_step(
     vae: AutoencoderKL,
     noise_scheduler: DDPMScheduler,
     unet: UNet2DConditionModel,
-- 
cgit v1.2.3-70-g09d2