소스 검색

Fix weighting config for SDXL v-pred

Fixes a small oversight I made.
catboxanon 9 달 전
부모
커밋
ac28cad998
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      configs/sd_xl_v.yaml

+ 1 - 1
configs/sd_xl_v.yaml

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