Merge pull request #16606 from AUTOMATIC1111/fix/vweighting

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