[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
6f887afb2b
commit
d54e2bc5f0
1 changed files with 0 additions and 1 deletions
|
|
@ -198,7 +198,6 @@ def _vae_conv_ndims(vae: Any) -> tuple[int, ...]:
|
|||
ndims: set[int] = set()
|
||||
try:
|
||||
from torch import nn
|
||||
|
||||
for module in vae.modules():
|
||||
if isinstance(module, nn.Conv3d):
|
||||
ndims.add(3)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue