Skip to content

Commit 0a4b843

Browse files
committed
Arm backend: Add ignores for missing imports in test/models/sta...
Some missing imports for third-party libraries used in stable_diffusion had ignores added to them. Signed-off-by: per.held@arm.com Change-Id: Icd86edd39ae4541813bc84c07fdab90571da98c2
1 parent d645543 commit 0a4b843

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,9 @@
77
from typing import Tuple
88

99
import torch
10-
from diffusers.models.transformers import SD3Transformer2DModel
10+
from diffusers.models.transformers import ( # type: ignore[import-not-found]
11+
SD3Transformer2DModel,
12+
)
1113

1214
from executorch.backends.arm.test import common
1315
from executorch.backends.arm.test.models.stable_diffusion.stable_diffusion_module_test_configs import (

backends/arm/test/models/stable_diffusion/test_vae_AutoencoderKL.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,12 @@
77
from typing import Tuple
88

99
import torch
10-
from diffusers.models.autoencoders import AutoencoderKL
11-
from diffusers.utils.testing_utils import floats_tensor
10+
from diffusers.models.autoencoders import ( # type: ignore[import-not-found]
11+
AutoencoderKL,
12+
)
13+
from diffusers.utils.testing_utils import ( # type: ignore[import-not-found]
14+
floats_tensor,
15+
)
1216

1317
from executorch.backends.arm.test import common
1418
from executorch.backends.arm.test.models.stable_diffusion.stable_diffusion_module_test_configs import (

0 commit comments

Comments
 (0)