Skip to content

Commit 4b843c8

Browse files
Fix variable name in docstring for PeftAdapterMixin.set_adapters (#13003)
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
1 parent d7a1c31 commit 4b843c8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/loaders/peft.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -478,7 +478,7 @@ def set_adapters(
478478
Args:
479479
adapter_names (`List[str]` or `str`):
480480
The names of the adapters to use.
481-
adapter_weights (`Union[List[float], float]`, *optional*):
481+
weights (`Union[List[float], float]`, *optional*):
482482
The adapter(s) weights to use with the UNet. If `None`, the weights are set to `1.0` for all the
483483
adapters.
484484
@@ -495,7 +495,7 @@ def set_adapters(
495495
"jbilcke-hf/sdxl-cinematic-1", weight_name="pytorch_lora_weights.safetensors", adapter_name="cinematic"
496496
)
497497
pipeline.load_lora_weights("nerijs/pixel-art-xl", weight_name="pixel-art-xl.safetensors", adapter_name="pixel")
498-
pipeline.unet.set_adapters(["cinematic", "pixel"], adapter_weights=[0.5, 0.5])
498+
pipeline.unet.set_adapters(["cinematic", "pixel"], weights=[0.5, 0.5])
499499
```
500500
"""
501501
if not USE_PEFT_BACKEND:

0 commit comments

Comments
 (0)