add missing kwarg

quantresearch1 · Oct 18, 2022 · 80e6a53 · 80e6a53
1 parent 99d8104
commit 80e6a53
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py
@@ -1470,6 +1470,7 @@ def __init__(
         self,
         concat_keys=("mask", "masked_image"),
         masked_image_key="masked_image",
+        finetune_keys=None,
         *args,
         **kwargs,
     ):