git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-04-21 06:06:02 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add XLabs IP-Adapter state dict for unit tests.

Browse Source
This commit is contained in:
Ryan Dick 2024-10-11 13:12:04 +00:00
parent 32c7cdd856
commit 7bf5927c43
1 changed files with 85 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
tests/backend/flux/ip_adapter/xlabs_flux_ip_adapter_state_dict.py Normal file
View File

@ -0,0 +1,85 @@
# State dict keys and shapes for an XLabs FLUX IP-Adapter model. Intended to be used for unit tests.
# These keys were extracted from:
# https://huggingface.co/XLabs-AI/flux-ip-adapter/blob/ad16be50d78a07ea83d8c4bde44ff9753235182e/flux-ip-adapter.safetensors
xlabs_sd_shapes = {
"double_blocks.0.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.0.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.0.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.0.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.1.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.1.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.1.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.1.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.10.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.10.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.10.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.10.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.11.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.11.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.11.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.11.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.12.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.12.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.12.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.12.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.13.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.13.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.13.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.13.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.14.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.14.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.14.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.14.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.15.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.15.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.15.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.15.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.16.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.16.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.16.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.16.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.17.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.17.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.17.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.17.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.18.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.18.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.18.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.18.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.2.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.2.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.2.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.2.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.3.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.3.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.3.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.3.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.4.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.4.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.4.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.4.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.5.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.5.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.5.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.5.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.6.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.6.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.6.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.6.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.7.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.7.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.7.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.7.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.8.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.8.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.8.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.8.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"double_blocks.9.processor.ip_adapter_double_stream_k_proj.bias": [3072],
"double_blocks.9.processor.ip_adapter_double_stream_k_proj.weight": [3072, 4096],
"double_blocks.9.processor.ip_adapter_double_stream_v_proj.bias": [3072],
"double_blocks.9.processor.ip_adapter_double_stream_v_proj.weight": [3072, 4096],
"ip_adapter_proj_model.norm.bias": [4096],
"ip_adapter_proj_model.norm.weight": [4096],
"ip_adapter_proj_model.proj.bias": [16384],
"ip_adapter_proj_model.proj.weight": [16384, 768],
}