Skip to content

Commit 2e387da

Browse files
yiyixuxuyiyixuxu
andauthored
fix IPAdapter unload_ip_adapter test (#6972)
add Co-authored-by: yiyixuxu <yixu310@gmail,com>
1 parent 9efe1e5 commit 2e387da

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/ip_adapters/test_ip_adapter_stable_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ def test_unload(self):
253253
pipeline.unload_ip_adapter()
254254

255255
assert getattr(pipeline, "image_encoder") is None
256-
assert getattr(pipeline, "feature_extractor") is None
256+
assert getattr(pipeline, "feature_extractor") is not None
257257
processors = [
258258
isinstance(attn_proc, (AttnProcessor, AttnProcessor2_0))
259259
for name, attn_proc in pipeline.unet.attn_processors.items()

0 commit comments

Comments
 (0)