Fix weighting config for SDXL v-pred

Fixes a small oversight I made.
This commit is contained in:
catboxanon
2024-10-29 11:49:09 -04:00
committed by GitHub
parent 5206b938ef
commit ac28cad998

View File

@@ -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: