Merge pull request #16606 from AUTOMATIC1111/fix/vweighting
Some checks are pending
Linter / ruff (push) Waiting to run
Linter / eslint (push) Waiting to run
Tests / tests on CPU with empty model (push) Waiting to run

Fix config for SDXL v-pred
This commit is contained in:
w-e-w
2024-10-30 09:51:41 +09:00
committed by GitHub

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