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 num_idx: 1000
weighting_config: weighting_config:
target: sgm.modules.diffusionmodules.denoiser_weighting.EpsWeighting target: sgm.modules.diffusionmodules.denoiser_weighting.VWeighting
scaling_config: scaling_config:
target: sgm.modules.diffusionmodules.denoiser_scaling.VScaling target: sgm.modules.diffusionmodules.denoiser_scaling.VScaling
discretization_config: discretization_config:
@@ -21,7 +21,7 @@ model:
params: params:
adm_in_channels: 2816 adm_in_channels: 2816
num_classes: sequential num_classes: sequential
use_checkpoint: True use_checkpoint: False
in_channels: 4 in_channels: 4
out_channels: 4 out_channels: 4
model_channels: 320 model_channels: 320