1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-24 20:59:27 +03:00
This commit is contained in:
sayakpaul
2023-12-11 18:22:20 +05:30
parent b868e8a2fc
commit c341111d69

View File

@@ -1327,6 +1327,7 @@ class StableDiffusionXLLoraLoaderMixin(LoraLoaderMixin):
unet_config=self.unet.config,
**kwargs,
)
print(f"Metadata: {metadata}")
is_correct_format = all("lora" in key for key in state_dict.keys())
if not is_correct_format:
raise ValueError("Invalid LoRA checkpoint.")