use_checkpoint = False

2025-08-03 19:02:27 +00:00 · 2024-05-15 15:20:40 -04:00
parent 1c0a0c4c26
commit 022d835565
8 changed files with 13 additions and 10 deletions
--- a/modules/sd_models_config.py
+++ b/modules/sd_models_config.py
@@ -35,7 +35,7 @@ def is_using_v_parameterization_for_sd2(state_dict):

    with sd_disable_initialization.DisableInitialization():
        unet = ldm.modules.diffusionmodules.openaimodel.UNetModel(
-            use_checkpoint=True,
+            use_checkpoint=False,
            use_fp16=False,
            image_size=32,
            in_channels=4,