From ac28cad998a5c6f09d1cd928f68d9f255ed72622 Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Tue, 29 Oct 2024 11:49:09 -0400
Subject: [PATCH 1/2] Fix weighting config for SDXL v-pred

Fixes a small oversight I made.
---
 configs/sd_xl_v.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/sd_xl_v.yaml b/configs/sd_xl_v.yaml
index c755dc74fda..5adbc256c01 100644
--- a/configs/sd_xl_v.yaml
+++ b/configs/sd_xl_v.yaml
@@ -10,7 +10,7 @@ model:
         num_idx: 1000
 
         weighting_config:
-          target: sgm.modules.diffusionmodules.denoiser_weighting.EpsWeighting
+          target: sgm.modules.diffusionmodules.denoiser_weighting.VWeighting
         scaling_config:
           target: sgm.modules.diffusionmodules.denoiser_scaling.VScaling
         discretization_config:

From e6f36d9cdc42d8008be482d449e85e737f3b989d Mon Sep 17 00:00:00 2001
From: catboxanon <122327233+catboxanon@users.noreply.github.com>
Date: Tue, 29 Oct 2024 13:27:32 -0400
Subject: [PATCH 2/2] sd_xl_v.yaml: use_checkpoint = False

In accordance with https://github.com/AUTOMATIC1111/stable-diffusion-webui/pull/15803
---
 configs/sd_xl_v.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/configs/sd_xl_v.yaml b/configs/sd_xl_v.yaml
index 5adbc256c01..9fbcbbac8b2 100644
--- a/configs/sd_xl_v.yaml
+++ b/configs/sd_xl_v.yaml
@@ -21,7 +21,7 @@ model:
       params:
         adm_in_channels: 2816
         num_classes: sequential
-        use_checkpoint: True
+        use_checkpoint: False
         in_channels: 4
         out_channels: 4
         model_channels: 320