diff --git a/ldm/models/diffusion/ddpm.py b/ldm/models/diffusion/ddpm.py index 4f71ba0..a7a74c2 100644 --- 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, ):