From d67c30512048e667cbf10c298a04c73f4eb5027c Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 3 Jan 2023 19:48:13 +0000 Subject: [PATCH] allow conversion from no state dict checkpoints --- scripts/convert_original_stable_diffusion_to_diffusers.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/convert_original_stable_diffusion_to_diffusers.py b/scripts/convert_original_stable_diffusion_to_diffusers.py index 1f42044954..9ccfc254b8 100644 --- a/scripts/convert_original_stable_diffusion_to_diffusers.py +++ b/scripts/convert_original_stable_diffusion_to_diffusers.py @@ -866,7 +866,9 @@ if __name__ == "__main__": else: print("global_step key not found in model") global_step = None - checkpoint = checkpoint["state_dict"] + + if "state_dict" in checkpoint: + checkpoint = checkpoint["state_dict"] upcast_attention = False if args.original_config_file is None: