Skip to content

Commit 9ef7b59

Browse files
committed
replace get_block_from_block_type to get_block
1 parent ca3ac4d commit 9ef7b59

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/models/autoencoders/autoencoder_dc.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
259259
return x
260260

261261

262-
def get_block_from_block_type(
262+
def get_block(
263263
block_type: str,
264264
in_channels: int,
265265
out_channels: int,
@@ -399,7 +399,7 @@ def __init__(
399399
stage = []
400400

401401
for _ in range(depth):
402-
block = get_block_from_block_type(
402+
block = get_block(
403403
block_type[stage_id],
404404
width,
405405
width,
@@ -485,7 +485,7 @@ def __init__(
485485
stage.append(upsample_block)
486486

487487
for _ in range(depth):
488-
block = get_block_from_block_type(
488+
block = get_block(
489489
block_type[stage_id],
490490
width,
491491
width,

0 commit comments

Comments
 (0)