Skip to content

Commit dbc6fcb

Browse files
authored
Merge branch 'main' into fix_docs_links
2 parents f1da19a + 04e783c commit dbc6fcb

File tree

6 files changed

+19
-4
lines changed

6 files changed

+19
-4
lines changed

docs/source/en/api/models/allegro_transformer3d.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
1818
```python
1919
from diffusers import AllegroTransformer3DModel
2020

21-
vae = AllegroTransformer3DModel.from_pretrained("rhymes-ai/Allegro", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
21+
transformer = AllegroTransformer3DModel.from_pretrained("rhymes-ai/Allegro", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
2222
```
2323

2424
## AllegroTransformer3DModel

docs/source/en/api/models/cogvideox_transformer3d.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
1818
```python
1919
from diffusers import CogVideoXTransformer3DModel
2020

21-
vae = CogVideoXTransformer3DModel.from_pretrained("THUDM/CogVideoX-2b", subfolder="transformer", torch_dtype=torch.float16).to("cuda")
21+
transformer = CogVideoXTransformer3DModel.from_pretrained("THUDM/CogVideoX-2b", subfolder="transformer", torch_dtype=torch.float16).to("cuda")
2222
```
2323

2424
## CogVideoXTransformer3DModel

docs/source/en/api/models/cogview3plus_transformer2d.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
1818
```python
1919
from diffusers import CogView3PlusTransformer2DModel
2020

21-
vae = CogView3PlusTransformer2DModel.from_pretrained("THUDM/CogView3Plus-3b", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
21+
transformer = CogView3PlusTransformer2DModel.from_pretrained("THUDM/CogView3Plus-3b", subfolder="transformer", torch_dtype=torch.bfloat16).to("cuda")
2222
```
2323

2424
## CogView3PlusTransformer2DModel

docs/source/en/api/models/mochi_transformer3d.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
1818
```python
1919
from diffusers import MochiTransformer3DModel
2020

21-
vae = MochiTransformer3DModel.from_pretrained("genmo/mochi-1-preview", subfolder="transformer", torch_dtype=torch.float16).to("cuda")
21+
transformer = MochiTransformer3DModel.from_pretrained("genmo/mochi-1-preview", subfolder="transformer", torch_dtype=torch.float16).to("cuda")
2222
```
2323

2424
## MochiTransformer3DModel

src/diffusers/loaders/unet.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -343,6 +343,17 @@ def _process_lora(
343343
else:
344344
if is_peft_version("<", "0.9.0"):
345345
lora_config_kwargs.pop("use_dora")
346+
347+
if "lora_bias" in lora_config_kwargs:
348+
if lora_config_kwargs["lora_bias"]:
349+
if is_peft_version("<=", "0.13.2"):
350+
raise ValueError(
351+
"You need `peft` 0.14.0 at least to use `bias` in LoRAs. Please upgrade your installation of `peft`."
352+
)
353+
else:
354+
if is_peft_version("<=", "0.13.2"):
355+
lora_config_kwargs.pop("lora_bias")
356+
346357
lora_config = LoraConfig(**lora_config_kwargs)
347358

348359
# adapter_name

tests/lora/test_lora_layers_sd3.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,11 @@
2929
from diffusers.utils import load_image
3030
from diffusers.utils.import_utils import is_accelerate_available
3131
from diffusers.utils.testing_utils import (
32+
nightly,
3233
numpy_cosine_similarity_distance,
3334
require_peft_backend,
3435
require_torch_gpu,
36+
slow,
3537
torch_device,
3638
)
3739

@@ -126,6 +128,8 @@ def test_modify_padding_mode(self):
126128
pass
127129

128130

131+
@slow
132+
@nightly
129133
@require_torch_gpu
130134
@require_peft_backend
131135
class LoraSD3IntegrationTests(unittest.TestCase):

0 commit comments

Comments
 (0)