update shift_factor

zRzRzRzRzRzRzR · zRzRzRzRzRzRzR · commit 29b0c81ea622 · 2025-02-24T17:06:43.000+08:00
diff --git a/scripts/convert_cogview4_to_diffusers_megatron.py b/scripts/convert_cogview4_to_diffusers_megatron.py
@@ -311,6 +311,7 @@ def main(args):
             "norm_num_groups": 32,
             "sample_size": 1024,
             "scaling_factor": 1.0,
+            "shift_factor": 0.0,
             "force_upcast": True,
             "use_quant_conv": False,
             "use_post_quant_conv": False,
