Merge pull request #16606 from AUTOMATIC1111/fix/vweighting

Fix config for SDXL v-pred
AUTOMATIC1111 · Oct 30, 2024 · aa52408 · aa52408
2 parents 28323cf + e6f36d9
commit aa52408
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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:
@@ -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