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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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: