1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
This commit is contained in:
patil-suraj
2023-09-07 19:27:40 +02:00
parent a10cd21b04
commit 5fa46f4b01

View File

@@ -971,7 +971,7 @@ class StableDiffusionXLAdapterPipeline(
return (image,)
return StableDiffusionXLPipelineOutput(images=image)
# Overrride to properly handle the loading and unloading of the additional text encoder.
def load_lora_weights(self, pretrained_model_name_or_path_or_dict: Union[str, Dict[str, torch.Tensor]], **kwargs):
# We could have accessed the unet config from `lora_state_dict()` too. We pass