Skip to content

GitLab

Unverified Commit 2e387dad authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix IPAdapter unload_ip_adapter test (#6972) 



add
Co-authored-by: default avataryiyixuxu <yixu310@gmail,com>
parent 9efe1e52
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/pipelines/ip_adapters/test_ip_adapter_stable_diffusion.py
View file @ 2e387dad
...@@ -253,7 +253,7 @@ class IPAdapterSDIntegrationTests(IPAdapterNightlyTestsMixin): ...@@ -253,7 +253,7 @@ class IPAdapterSDIntegrationTests(IPAdapterNightlyTestsMixin):
pipeline.unload_ip_adapter() pipeline.unload_ip_adapter()
assert getattr(pipeline, "image_encoder") is None assert getattr(pipeline, "image_encoder") is None
assert getattr(pipeline, "feature_extractor") is None assert getattr(pipeline, "feature_extractor") is not None
processors = [ processors = [
isinstance(attn_proc, (AttnProcessor, AttnProcessor2_0)) isinstance(attn_proc, (AttnProcessor, AttnProcessor2_0))
for name, attn_proc in pipeline.unet.attn_processors.items() for name, attn_proc in pipeline.unet.attn_processors.items()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment