feature: support sdxl (#431)
- adds support for (SDXL)[https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0] - adds sliced encoding/decoding to refiners sdxl pipeline - doesn't support inpainting, controlnets - monkeypatches self_attention_guidance to use sliced attention - adds a bunch of model weight translation utilities and weightmaps - add [opendalle 1.1](https://huggingface.co/dataautogpt3/OpenDalleV1.1) - change default model to opendalle - fix: better handle special characters in path inputs on command line **todo** - add testspull/433/head
parent
fb7cff3684
commit
42a045e8e6
@ -0,0 +1,206 @@
|
||||
import logging
|
||||
import re
|
||||
from dataclasses import asdict, dataclass, field
|
||||
from typing import Dict
|
||||
|
||||
import torch
|
||||
from refiners.fluxion import load_from_safetensors
|
||||
from torch import device as Device
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
TensorDict = Dict[str, torch.Tensor]
|
||||
|
||||
|
||||
@dataclass
|
||||
class WeightTranslationMap:
|
||||
name_map: dict[str, str | None] = field(default_factory=dict)
|
||||
regex_map: dict[str, str | None] = field(default_factory=dict)
|
||||
ignore_prefixes: list[str] = field(default_factory=list)
|
||||
source_aliases: dict[str, str] = field(default_factory=dict)
|
||||
reshapes: dict[str, tuple[int, ...]] = field(default_factory=dict)
|
||||
|
||||
def load_and_translate_weights(
|
||||
self, source_path: str, device: Device | str = "cpu"
|
||||
) -> TensorDict:
|
||||
extension = source_path.split(".")[-1]
|
||||
if extension in ["pth", "pt", "bin"]:
|
||||
source_weights = torch.load(source_path, map_location="cpu")
|
||||
|
||||
elif extension in ["safetensors"]:
|
||||
source_weights = load_from_safetensors(source_path, device=device)
|
||||
else:
|
||||
msg = f"Unsupported extension {extension}"
|
||||
raise ValueError(msg)
|
||||
|
||||
return self.translate_weights(source_weights)
|
||||
|
||||
def translate_weights(self, source_weights: TensorDict) -> TensorDict:
|
||||
return translate_weights(source_weights=source_weights, weight_map=self)
|
||||
|
||||
def add_regex_replacer(self, find_pattern: str, replace_pattern: str):
|
||||
orig_size = len(self.name_map)
|
||||
# see which keys of name_map
|
||||
find_pattern_c = re.compile(find_pattern)
|
||||
matched_keys = [k for k in self.name_map if find_pattern_c.match(k)]
|
||||
# print(f"Pattern {find_pattern} matched {len(matched_keys)} keys")
|
||||
for k in matched_keys:
|
||||
match = find_pattern_c.match(k)
|
||||
if not match:
|
||||
continue
|
||||
data = match.groupdict()
|
||||
new_k = render_fstring(replace_pattern, data)
|
||||
expected_k = self.name_map[k]
|
||||
assert new_k == expected_k
|
||||
# print(f"Replacing {k} with {new_k}")
|
||||
del self.name_map[k]
|
||||
|
||||
self.regex_map[find_pattern] = replace_pattern
|
||||
print(
|
||||
f"Adding pattern reduced name_map from {orig_size} to {len(self.name_map)}"
|
||||
)
|
||||
|
||||
def save(self, path):
|
||||
import json
|
||||
import os
|
||||
|
||||
os.makedirs(os.path.dirname(path), exist_ok=True)
|
||||
|
||||
with open(path, "w") as f:
|
||||
json.dump(asdict(self), f, indent=4)
|
||||
|
||||
@classmethod
|
||||
def load(cls, path):
|
||||
import json
|
||||
|
||||
with open(path) as f:
|
||||
d = json.load(f)
|
||||
return cls(**d)
|
||||
|
||||
|
||||
def translate_weights(
|
||||
source_weights: TensorDict, weight_map: WeightTranslationMap
|
||||
) -> TensorDict:
|
||||
new_state_dict: TensorDict = {}
|
||||
# print(f"Translating {len(source_weights)} weights")
|
||||
# print(f"Using {len(weight_map.name_map)} name mappings")
|
||||
# print(source_weights.keys())
|
||||
|
||||
source_weights = flatten_dict(source_weights)
|
||||
|
||||
for source_key in list(source_weights.keys()):
|
||||
source_key = weight_map.source_aliases.get(source_key, source_key)
|
||||
try:
|
||||
target_key = weight_map.name_map[source_key]
|
||||
# print(f"Found {source_prefix} -> {target_prefix}")
|
||||
except KeyError:
|
||||
continue
|
||||
if target_key is None:
|
||||
# mapped to None means we ignore it
|
||||
source_weights.pop(source_key)
|
||||
else:
|
||||
# print(f"Adding {target_key}")
|
||||
new_state_dict[target_key] = source_weights.pop(source_key)
|
||||
|
||||
for source_key in list(source_weights.keys()):
|
||||
try:
|
||||
source_prefix, suffix = source_key.rsplit(sep=".", maxsplit=1)
|
||||
except ValueError:
|
||||
# no dots
|
||||
continue
|
||||
# print(f"Checking {source_prefix} {suffix}")
|
||||
|
||||
source_prefix = weight_map.source_aliases.get(source_prefix, source_prefix)
|
||||
try:
|
||||
target_prefix = weight_map.name_map[source_prefix]
|
||||
# print(f"Found {source_prefix} -> {target_prefix}")
|
||||
except KeyError:
|
||||
continue
|
||||
if target_prefix is None:
|
||||
# mapped to None means we ignore it
|
||||
source_weights.pop(source_key)
|
||||
continue
|
||||
else:
|
||||
target_key = ".".join([target_prefix, suffix])
|
||||
# print(f"Adding {target_key}")
|
||||
new_state_dict[target_key] = source_weights.pop(source_key)
|
||||
|
||||
for source_key in list(source_weights.keys()):
|
||||
try:
|
||||
source_prefix, suffix = source_key.rsplit(sep=".", maxsplit=1)
|
||||
except ValueError:
|
||||
# no dots
|
||||
continue
|
||||
for pattern, replace_pattern in weight_map.regex_map.items():
|
||||
match = re.match(pattern, source_prefix)
|
||||
if match:
|
||||
match_data = match.groupdict()
|
||||
new_k = render_fstring(replace_pattern, match_data)
|
||||
new_k = ".".join([new_k, suffix])
|
||||
new_state_dict[new_k] = source_weights.pop(source_key)
|
||||
|
||||
if source_weights:
|
||||
msg = f"Unmapped keys: {list(source_weights.keys())}"
|
||||
print(msg)
|
||||
for k in source_weights:
|
||||
if isinstance(source_weights[k], torch.Tensor):
|
||||
print(f" {k}: {source_weights[k].shape}")
|
||||
else:
|
||||
print(f" {k}: {repr(source_weights[k])[:100]}")
|
||||
|
||||
if weight_map.reshapes:
|
||||
for key, new_shape in weight_map.reshapes.items():
|
||||
if key in new_state_dict:
|
||||
new_state_dict[key] = new_state_dict[key].reshape(new_shape)
|
||||
|
||||
return new_state_dict
|
||||
|
||||
|
||||
def flatten_dict(d, sep="."):
|
||||
"""
|
||||
Flattens a nested dictionary into a dictionary with dot-separated keys.
|
||||
This function removes items from the original dictionary as they are added to the new one.
|
||||
The function uses an iterative approach instead of recursion.
|
||||
|
||||
Parameters:
|
||||
d (dict): The dictionary to flatten.
|
||||
sep (str): The separator to use between keys.
|
||||
|
||||
Returns:
|
||||
dict: A flattened dictionary.
|
||||
"""
|
||||
flat_dict = {}
|
||||
stack = [("", d)]
|
||||
|
||||
while stack:
|
||||
parent_key, current_dict = stack.pop()
|
||||
keys = list(current_dict.keys()) # Create a list of keys to avoid RuntimeError
|
||||
for k in keys:
|
||||
new_key = f"{parent_key}{sep}{k}" if parent_key else k
|
||||
if isinstance(current_dict[k], dict):
|
||||
stack.append((new_key, current_dict[k]))
|
||||
else:
|
||||
flat_dict[new_key] = current_dict.pop(k)
|
||||
|
||||
return flat_dict
|
||||
|
||||
|
||||
def render_fstring(fstring, variables):
|
||||
"""
|
||||
Render a string formatted like an f-string using the provided variables.
|
||||
|
||||
DANGER: This is a security risk if the fstring is user-provided.
|
||||
|
||||
Args:
|
||||
fstring (str): The template string with placeholders for variables.
|
||||
variables (dict): A dictionary containing the variables to be used in the f-string.
|
||||
|
||||
Returns:
|
||||
str: The rendered string with variables substituted.
|
||||
"""
|
||||
# Use locals().update to add the variables to the local scope
|
||||
locals().update(variables)
|
||||
|
||||
# Evaluate the f-string using eval with an f-string formatted string
|
||||
return eval(f'f"""{fstring}"""')
|
@ -0,0 +1,115 @@
|
||||
import os
|
||||
from functools import lru_cache
|
||||
|
||||
from torch import device as Device
|
||||
|
||||
from imaginairy.weight_management.translation import TensorDict, WeightTranslationMap
|
||||
|
||||
_current_folder = os.path.dirname(os.path.abspath(__file__))
|
||||
weight_map_folder = os.path.join(_current_folder, "weight_maps")
|
||||
|
||||
|
||||
@lru_cache
|
||||
def _weight_map(map_name: str) -> WeightTranslationMap:
|
||||
map_path = os.path.join(weight_map_folder, f"{map_name}.weightmap.json")
|
||||
return WeightTranslationMap.load(map_path)
|
||||
|
||||
|
||||
def transformers_text_encoder_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Transformers-ClipTextEncoder")
|
||||
|
||||
|
||||
def transformers_image_encoder_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Transformers-ClipImageEncoder-SD21")
|
||||
|
||||
|
||||
def diffusers_autoencoder_kl_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-AutoencoderKL-SD")
|
||||
|
||||
|
||||
def diffusers_unet_sd15_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-UNet-SD15")
|
||||
|
||||
|
||||
def diffusers_unet_sdxl_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-UNet-SDXL")
|
||||
|
||||
|
||||
def informative_drawings_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("InformativeDrawings")
|
||||
|
||||
|
||||
def diffusers_controlnet_sd15_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-Controlnet-SD15")
|
||||
|
||||
|
||||
def diffusers_ip_adapter_sd15_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-IPAdapter-SD15")
|
||||
|
||||
|
||||
def diffusers_ip_adapter_sdxl_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-IPAdapter-SDXL")
|
||||
|
||||
|
||||
def diffusers_ip_adapter_plus_sd15_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-IPAdapterPlus-SD15")
|
||||
|
||||
|
||||
def diffusers_ip_adapter_plus_sdxl_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-IPAdapterPlus-SDXL")
|
||||
|
||||
|
||||
def diffusers_t2i_adapter_sd15_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-T2IAdapter-SD15")
|
||||
|
||||
|
||||
def diffusers_t2i_adapter_sdxl_to_refiners_translator() -> WeightTranslationMap:
|
||||
return _weight_map("Diffusers-T2IAdapter-SDXL")
|
||||
|
||||
|
||||
class DoubleTextEncoderTranslator:
|
||||
def __init__(self):
|
||||
self.translator = transformers_text_encoder_to_refiners_translator()
|
||||
|
||||
def load_and_translate_weights(
|
||||
self,
|
||||
text_encoder_l_weights_path: str,
|
||||
text_encoder_g_weights_path: str,
|
||||
device: Device | str = "cpu",
|
||||
) -> TensorDict:
|
||||
text_encoder_l_weights = self.translator.load_and_translate_weights(
|
||||
text_encoder_l_weights_path, device=device
|
||||
)
|
||||
text_encoder_g_weights = self.translator.load_and_translate_weights(
|
||||
text_encoder_g_weights_path, device=device
|
||||
)
|
||||
return self.translate_weights(text_encoder_l_weights, text_encoder_g_weights)
|
||||
|
||||
def translate_weights(
|
||||
self, text_encoder_l_weights: TensorDict, text_encoder_g_weights: TensorDict
|
||||
) -> TensorDict:
|
||||
new_sd: TensorDict = {}
|
||||
|
||||
for k in list(text_encoder_l_weights.keys()):
|
||||
if k.startswith("TransformerLayer_12"):
|
||||
text_encoder_l_weights.pop(k)
|
||||
elif k.startswith("LayerNorm"):
|
||||
text_encoder_l_weights.pop(k)
|
||||
else:
|
||||
new_key = f"Parallel.CLIPTextEncoderL.{k}"
|
||||
new_sd[new_key] = text_encoder_l_weights.pop(k)
|
||||
|
||||
new_sd[
|
||||
"Parallel.TextEncoderWithPooling.Parallel.Chain.Linear.weight"
|
||||
] = text_encoder_g_weights.pop("Linear.weight")
|
||||
for k in list(text_encoder_g_weights.keys()):
|
||||
if k.startswith("TransformerLayer_32"):
|
||||
new_key = f"Parallel.TextEncoderWithPooling.Parallel.Chain.CLIPTextEncoderG.TransformerLayer{k[19:]}"
|
||||
elif k.startswith("LayerNorm"):
|
||||
new_key = f"Parallel.TextEncoderWithPooling.Parallel.Chain.CLIPTextEncoderG.{k}"
|
||||
else:
|
||||
new_key = f"Parallel.TextEncoderWithPooling.CLIPTextEncoderG.{k}"
|
||||
|
||||
new_sd[new_key] = text_encoder_g_weights.pop(k)
|
||||
|
||||
return new_sd
|
@ -0,0 +1,174 @@
|
||||
{
|
||||
"name_map": {
|
||||
"encoder.conv_in": "Encoder.Conv2d",
|
||||
"encoder.down_blocks.0.resnets.0.norm1": "Encoder.Chain_1.Chain_1.Resnet_1.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.0.resnets.0.norm2": "Encoder.Chain_1.Chain_1.Resnet_1.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.0.resnets.1.norm1": "Encoder.Chain_1.Chain_1.Resnet_2.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.0.resnets.1.norm2": "Encoder.Chain_1.Chain_1.Resnet_2.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.1.resnets.0.norm1": "Encoder.Chain_1.Chain_2.Resnet_1.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.3.resnets.0.norm2": "Decoder.Chain_1.Chain_5.Resnet_1.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.3.resnets.1.norm1": "Decoder.Chain_1.Chain_5.Resnet_2.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.3.resnets.1.norm2": "Decoder.Chain_1.Chain_5.Resnet_2.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.3.resnets.2.norm1": "Decoder.Chain_1.Chain_5.Resnet_3.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.3.resnets.2.norm2": "Decoder.Chain_1.Chain_5.Resnet_3.Chain.GroupNorm_2",
|
||||
"decoder.conv_norm_out": "Decoder.Chain_2.GroupNorm",
|
||||
"encoder.down_blocks.0.resnets.0.conv1": "Encoder.Chain_1.Chain_1.Resnet_1.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.0.resnets.0.conv2": "Encoder.Chain_1.Chain_1.Resnet_1.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.0.resnets.1.conv1": "Encoder.Chain_1.Chain_1.Resnet_2.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.0.resnets.1.conv2": "Encoder.Chain_1.Chain_1.Resnet_2.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.0.downsamplers.0.conv": "Encoder.Chain_1.Chain_1.Downsample.Conv2d",
|
||||
"decoder.up_blocks.3.resnets.0.conv2": "Decoder.Chain_1.Chain_5.Resnet_1.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.3.resnets.1.conv1": "Decoder.Chain_1.Chain_5.Resnet_2.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.3.resnets.1.conv2": "Decoder.Chain_1.Chain_5.Resnet_2.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.3.resnets.2.conv1": "Decoder.Chain_1.Chain_5.Resnet_3.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.3.resnets.2.conv2": "Decoder.Chain_1.Chain_5.Resnet_3.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.1.resnets.0.conv1": "Encoder.Chain_1.Chain_2.Resnet_1.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.1.resnets.0.norm2": "Encoder.Chain_1.Chain_2.Resnet_1.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.1.resnets.1.norm1": "Encoder.Chain_1.Chain_2.Resnet_2.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.1.resnets.1.norm2": "Encoder.Chain_1.Chain_2.Resnet_2.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.2.resnets.0.norm1": "Encoder.Chain_1.Chain_3.Resnet_1.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.2.resnets.0.norm2": "Decoder.Chain_1.Chain_4.Resnet_1.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.2.resnets.1.norm1": "Decoder.Chain_1.Chain_4.Resnet_2.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.2.resnets.1.norm2": "Decoder.Chain_1.Chain_4.Resnet_2.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.2.resnets.2.norm1": "Decoder.Chain_1.Chain_4.Resnet_3.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.2.resnets.2.norm2": "Decoder.Chain_1.Chain_4.Resnet_3.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.3.resnets.0.norm1": "Decoder.Chain_1.Chain_5.Resnet_1.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.1.resnets.0.conv2": "Encoder.Chain_1.Chain_2.Resnet_1.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.1.resnets.1.conv1": "Encoder.Chain_1.Chain_2.Resnet_2.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.1.resnets.1.conv2": "Encoder.Chain_1.Chain_2.Resnet_2.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.1.downsamplers.0.conv": "Encoder.Chain_1.Chain_2.Downsample.Conv2d",
|
||||
"decoder.up_blocks.2.resnets.0.conv2": "Decoder.Chain_1.Chain_4.Resnet_1.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.2.resnets.1.conv1": "Decoder.Chain_1.Chain_4.Resnet_2.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.2.resnets.1.conv2": "Decoder.Chain_1.Chain_4.Resnet_2.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.2.resnets.2.conv1": "Decoder.Chain_1.Chain_4.Resnet_3.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.2.resnets.2.conv2": "Decoder.Chain_1.Chain_4.Resnet_3.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.2.upsamplers.0.conv": "Decoder.Chain_1.Chain_4.Upsample.Conv2d",
|
||||
"encoder.down_blocks.1.resnets.0.conv_shortcut": "Encoder.Chain_1.Chain_2.Resnet_1.Conv2d",
|
||||
"encoder.down_blocks.2.resnets.0.conv1": "Encoder.Chain_1.Chain_3.Resnet_1.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.2.resnets.0.norm2": "Encoder.Chain_1.Chain_3.Resnet_1.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.2.resnets.1.norm1": "Encoder.Chain_1.Chain_3.Resnet_2.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.2.resnets.1.norm2": "Encoder.Chain_1.Chain_3.Resnet_2.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.3.resnets.0.norm1": "Encoder.Chain_1.Chain_4.Resnet_1.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.3.resnets.0.norm2": "Encoder.Chain_1.Chain_4.Resnet_1.Chain.GroupNorm_2",
|
||||
"encoder.down_blocks.3.resnets.1.norm1": "Encoder.Chain_1.Chain_4.Resnet_2.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.3.resnets.1.norm2": "Encoder.Chain_1.Chain_4.Resnet_2.Chain.GroupNorm_2",
|
||||
"encoder.mid_block.resnets.0.norm1": "Encoder.Chain_1.Chain_5.Resnet_1.Chain.GroupNorm_1",
|
||||
"encoder.mid_block.resnets.0.norm2": "Encoder.Chain_1.Chain_5.Resnet_1.Chain.GroupNorm_2",
|
||||
"encoder.mid_block.attentions.0.group_norm": "Encoder.Chain_1.Chain_5.Residual.GroupNorm",
|
||||
"encoder.mid_block.resnets.1.norm1": "Encoder.Chain_1.Chain_5.Resnet_2.Chain.GroupNorm_1",
|
||||
"encoder.mid_block.resnets.1.norm2": "Encoder.Chain_1.Chain_5.Resnet_2.Chain.GroupNorm_2",
|
||||
"encoder.conv_norm_out": "Encoder.Chain_2.GroupNorm",
|
||||
"decoder.mid_block.resnets.0.norm1": "Decoder.Chain_1.Chain_1.Resnet_1.Chain.GroupNorm_1",
|
||||
"decoder.mid_block.resnets.0.norm2": "Decoder.Chain_1.Chain_1.Resnet_1.Chain.GroupNorm_2",
|
||||
"decoder.mid_block.attentions.0.group_norm": "Decoder.Chain_1.Chain_1.Residual.GroupNorm",
|
||||
"decoder.mid_block.resnets.1.norm1": "Decoder.Chain_1.Chain_1.Resnet_2.Chain.GroupNorm_1",
|
||||
"decoder.mid_block.resnets.1.norm2": "Decoder.Chain_1.Chain_1.Resnet_2.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.0.resnets.0.norm1": "Decoder.Chain_1.Chain_2.Resnet_1.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.0.resnets.0.norm2": "Decoder.Chain_1.Chain_2.Resnet_1.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.0.resnets.1.norm1": "Decoder.Chain_1.Chain_2.Resnet_2.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.0.resnets.1.norm2": "Decoder.Chain_1.Chain_2.Resnet_2.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.0.resnets.2.norm1": "Decoder.Chain_1.Chain_2.Resnet_3.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.0.resnets.2.norm2": "Decoder.Chain_1.Chain_2.Resnet_3.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.1.resnets.0.norm1": "Decoder.Chain_1.Chain_3.Resnet_1.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.1.resnets.0.norm2": "Decoder.Chain_1.Chain_3.Resnet_1.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.1.resnets.1.norm1": "Decoder.Chain_1.Chain_3.Resnet_2.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.1.resnets.1.norm2": "Decoder.Chain_1.Chain_3.Resnet_2.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.1.resnets.2.norm1": "Decoder.Chain_1.Chain_3.Resnet_3.Chain.GroupNorm_1",
|
||||
"decoder.up_blocks.1.resnets.2.norm2": "Decoder.Chain_1.Chain_3.Resnet_3.Chain.GroupNorm_2",
|
||||
"decoder.up_blocks.2.resnets.0.norm1": "Decoder.Chain_1.Chain_4.Resnet_1.Chain.GroupNorm_1",
|
||||
"encoder.down_blocks.2.resnets.0.conv2": "Encoder.Chain_1.Chain_3.Resnet_1.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.2.resnets.1.conv1": "Encoder.Chain_1.Chain_3.Resnet_2.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.2.resnets.1.conv2": "Encoder.Chain_1.Chain_3.Resnet_2.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.2.downsamplers.0.conv": "Encoder.Chain_1.Chain_3.Downsample.Conv2d",
|
||||
"encoder.down_blocks.3.resnets.0.conv1": "Encoder.Chain_1.Chain_4.Resnet_1.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.3.resnets.0.conv2": "Encoder.Chain_1.Chain_4.Resnet_1.Chain.Conv2d_2",
|
||||
"encoder.down_blocks.3.resnets.1.conv1": "Encoder.Chain_1.Chain_4.Resnet_2.Chain.Conv2d_1",
|
||||
"encoder.down_blocks.3.resnets.1.conv2": "Encoder.Chain_1.Chain_4.Resnet_2.Chain.Conv2d_2",
|
||||
"encoder.mid_block.resnets.0.conv1": "Encoder.Chain_1.Chain_5.Resnet_1.Chain.Conv2d_1",
|
||||
"encoder.mid_block.resnets.0.conv2": "Encoder.Chain_1.Chain_5.Resnet_1.Chain.Conv2d_2",
|
||||
"encoder.mid_block.resnets.1.conv1": "Encoder.Chain_1.Chain_5.Resnet_2.Chain.Conv2d_1",
|
||||
"encoder.mid_block.resnets.1.conv2": "Encoder.Chain_1.Chain_5.Resnet_2.Chain.Conv2d_2",
|
||||
"decoder.mid_block.resnets.0.conv1": "Decoder.Chain_1.Chain_1.Resnet_1.Chain.Conv2d_1",
|
||||
"decoder.mid_block.resnets.0.conv2": "Decoder.Chain_1.Chain_1.Resnet_1.Chain.Conv2d_2",
|
||||
"decoder.mid_block.resnets.1.conv1": "Decoder.Chain_1.Chain_1.Resnet_2.Chain.Conv2d_1",
|
||||
"decoder.mid_block.resnets.1.conv2": "Decoder.Chain_1.Chain_1.Resnet_2.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.0.resnets.0.conv1": "Decoder.Chain_1.Chain_2.Resnet_1.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.0.resnets.0.conv2": "Decoder.Chain_1.Chain_2.Resnet_1.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.0.resnets.1.conv1": "Decoder.Chain_1.Chain_2.Resnet_2.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.0.resnets.1.conv2": "Decoder.Chain_1.Chain_2.Resnet_2.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.0.resnets.2.conv1": "Decoder.Chain_1.Chain_2.Resnet_3.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.0.resnets.2.conv2": "Decoder.Chain_1.Chain_2.Resnet_3.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.0.upsamplers.0.conv": "Decoder.Chain_1.Chain_2.Upsample.Conv2d",
|
||||
"decoder.up_blocks.1.resnets.0.conv1": "Decoder.Chain_1.Chain_3.Resnet_1.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.1.resnets.0.conv2": "Decoder.Chain_1.Chain_3.Resnet_1.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.1.resnets.1.conv1": "Decoder.Chain_1.Chain_3.Resnet_2.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.1.resnets.1.conv2": "Decoder.Chain_1.Chain_3.Resnet_2.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.1.resnets.2.conv1": "Decoder.Chain_1.Chain_3.Resnet_3.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.1.resnets.2.conv2": "Decoder.Chain_1.Chain_3.Resnet_3.Chain.Conv2d_2",
|
||||
"decoder.up_blocks.1.upsamplers.0.conv": "Decoder.Chain_1.Chain_3.Upsample.Conv2d",
|
||||
"encoder.down_blocks.2.resnets.0.conv_shortcut": "Encoder.Chain_1.Chain_3.Resnet_1.Conv2d",
|
||||
"encoder.mid_block.attentions.0.to_q": "Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Distribute.Linear_1",
|
||||
"encoder.mid_block.attentions.0.to_k": "Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Distribute.Linear_2",
|
||||
"encoder.mid_block.attentions.0.to_v": "Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Distribute.Linear_3",
|
||||
"encoder.mid_block.attentions.0.to_out.0": "Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Linear",
|
||||
"decoder.mid_block.attentions.0.to_q": "Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Distribute.Linear_1",
|
||||
"decoder.mid_block.attentions.0.to_k": "Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Distribute.Linear_2",
|
||||
"decoder.mid_block.attentions.0.to_v": "Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Distribute.Linear_3",
|
||||
"decoder.mid_block.attentions.0.to_out.0": "Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Linear",
|
||||
"encoder.conv_out": "Encoder.Chain_2.Conv2d",
|
||||
"quant_conv": "Encoder.Chain_3.Conv2d",
|
||||
"post_quant_conv": "Decoder.Conv2d_1",
|
||||
"decoder.conv_in": "Decoder.Conv2d_2",
|
||||
"decoder.up_blocks.2.resnets.0.conv1": "Decoder.Chain_1.Chain_4.Resnet_1.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.2.resnets.0.conv_shortcut": "Decoder.Chain_1.Chain_4.Resnet_1.Conv2d",
|
||||
"decoder.up_blocks.3.resnets.0.conv1": "Decoder.Chain_1.Chain_5.Resnet_1.Chain.Conv2d_1",
|
||||
"decoder.up_blocks.3.resnets.0.conv_shortcut": "Decoder.Chain_1.Chain_5.Resnet_1.Conv2d",
|
||||
"decoder.conv_out": "Decoder.Chain_2.Conv2d"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {
|
||||
"encoder.mid_block.attentions.0.value": "encoder.mid_block.attentions.0.to_v",
|
||||
"decoder.mid_block.attentions.0.value": "decoder.mid_block.attentions.0.to_v",
|
||||
"decoder.mid_block.attentions.0.proj_attn": "decoder.mid_block.attentions.0.to_out.0",
|
||||
"encoder.mid_block.attentions.0.proj_attn": "encoder.mid_block.attentions.0.to_out.0",
|
||||
"encoder.mid_block.attentions.0.key": "encoder.mid_block.attentions.0.to_k",
|
||||
"decoder.mid_block.attentions.0.key": "decoder.mid_block.attentions.0.to_k",
|
||||
"decoder.mid_block.attentions.0.query": "decoder.mid_block.attentions.0.to_q",
|
||||
"encoder.mid_block.attentions.0.query": "encoder.mid_block.attentions.0.to_q"
|
||||
},
|
||||
"reshapes": {
|
||||
"Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Distribute.Linear_1.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Distribute.Linear_2.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Distribute.Linear_3.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Encoder.Chain_1.Chain_5.Residual.SelfAttention2d.Linear.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Distribute.Linear_1.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Distribute.Linear_2.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Distribute.Linear_3.weight": [
|
||||
512,
|
||||
512
|
||||
],
|
||||
"Decoder.Chain_1.Chain_1.Residual.SelfAttention2d.Linear.weight": [
|
||||
512,
|
||||
512
|
||||
]
|
||||
}
|
||||
}
|
@ -0,0 +1,199 @@
|
||||
{
|
||||
"name_map": {
|
||||
"time_embedding.linear_1": "TimestepEncoder.RangeEncoder.Linear_1",
|
||||
"time_embedding.linear_2": "TimestepEncoder.RangeEncoder.Linear_2",
|
||||
"down_blocks.2.resnets.0.time_emb_proj": "DownBlocks.Chain_8.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.2.resnets.1.time_emb_proj": "DownBlocks.Chain_9.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.3.resnets.0.time_emb_proj": "DownBlocks.Chain_11.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.3.resnets.1.time_emb_proj": "DownBlocks.Chain_12.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"mid_block.resnets.0.time_emb_proj": "MiddleBlock.ResidualBlock_1.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"mid_block.resnets.1.time_emb_proj": "MiddleBlock.ResidualBlock_2.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"conv_in": "DownBlocks.Chain_1.Conv2d",
|
||||
"controlnet_cond_embedding.conv_in": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_1.Conv2d",
|
||||
"controlnet_cond_embedding.blocks.0": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_2.Conv2d_1",
|
||||
"controlnet_cond_embedding.blocks.1": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_2.Conv2d_2",
|
||||
"controlnet_cond_embedding.blocks.2": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_3.Conv2d_1",
|
||||
"controlnet_cond_embedding.blocks.3": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_3.Conv2d_2",
|
||||
"controlnet_cond_embedding.blocks.4": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_4.Conv2d_1",
|
||||
"controlnet_cond_embedding.blocks.5": "DownBlocks.Chain_1.Residual.ConditionEncoder.Chain_4.Conv2d_2",
|
||||
"controlnet_cond_embedding.conv_out": "DownBlocks.Chain_1.Residual.ConditionEncoder.Conv2d",
|
||||
"down_blocks.0.resnets.0.norm1": "DownBlocks.Chain_2.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.0.resnets.0.norm2": "DownBlocks.Chain_2.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.0.attentions.0.norm": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.0.resnets.1.norm1": "DownBlocks.Chain_3.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.0.resnets.1.norm2": "DownBlocks.Chain_3.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.0.attentions.1.norm": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.1.resnets.0.norm1": "DownBlocks.Chain_5.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.0.resnets.0.conv1": "DownBlocks.Chain_2.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.0.resnets.0.conv2": "DownBlocks.Chain_2.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.0.resnets.1.conv1": "DownBlocks.Chain_3.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.0.resnets.1.conv2": "DownBlocks.Chain_3.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.0.downsamplers.0.conv": "DownBlocks.Chain_4.Downsample.Conv2d",
|
||||
"down_blocks.0.resnets.0.time_emb_proj": "DownBlocks.Chain_2.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.0.resnets.1.time_emb_proj": "DownBlocks.Chain_3.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"controlnet_down_blocks.0": "DownBlocks.Chain_1.Passthrough.Conv2d",
|
||||
"down_blocks.0.attentions.0.proj_in": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.0.attentions.0.proj_out": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_down_blocks.1": "DownBlocks.Chain_2.Passthrough.Conv2d",
|
||||
"down_blocks.0.attentions.1.proj_in": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.0.attentions.1.proj_out": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_down_blocks.2": "DownBlocks.Chain_3.Passthrough.Conv2d",
|
||||
"controlnet_down_blocks.3": "DownBlocks.Chain_4.Passthrough.Conv2d",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.norm1": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.norm2": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.norm3": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.norm1": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.norm2": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.norm3": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.1.resnets.0.conv1": "DownBlocks.Chain_5.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.1.resnets.0.time_emb_proj": "DownBlocks.Chain_5.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.1.resnets.1.time_emb_proj": "DownBlocks.Chain_6.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.1.resnets.0.norm2": "DownBlocks.Chain_5.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.1.attentions.0.norm": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.1.resnets.1.norm1": "DownBlocks.Chain_6.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.1.resnets.1.norm2": "DownBlocks.Chain_6.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.1.attentions.1.norm": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.2.resnets.0.norm1": "DownBlocks.Chain_8.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.1.resnets.0.conv2": "DownBlocks.Chain_5.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.1.resnets.1.conv1": "DownBlocks.Chain_6.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.1.resnets.1.conv2": "DownBlocks.Chain_6.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.1.downsamplers.0.conv": "DownBlocks.Chain_7.Downsample.Conv2d",
|
||||
"down_blocks.1.resnets.0.conv_shortcut": "DownBlocks.Chain_5.ResidualBlock.Conv2d",
|
||||
"down_blocks.1.attentions.0.proj_in": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.1.attentions.0.proj_out": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_down_blocks.4": "DownBlocks.Chain_5.Passthrough.Conv2d",
|
||||
"down_blocks.1.attentions.1.proj_in": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.1.attentions.1.proj_out": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_down_blocks.5": "DownBlocks.Chain_6.Passthrough.Conv2d",
|
||||
"controlnet_down_blocks.6": "DownBlocks.Chain_7.Passthrough.Conv2d",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.norm1": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.norm2": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.norm3": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.norm1": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.norm2": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.norm3": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.2.resnets.0.conv1": "DownBlocks.Chain_8.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.2.resnets.0.norm2": "DownBlocks.Chain_8.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.2.attentions.0.norm": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.2.resnets.1.norm1": "DownBlocks.Chain_9.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.2.resnets.1.norm2": "DownBlocks.Chain_9.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.2.attentions.1.norm": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.3.resnets.0.norm1": "DownBlocks.Chain_11.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.3.resnets.0.norm2": "DownBlocks.Chain_11.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.3.resnets.1.norm1": "DownBlocks.Chain_12.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.3.resnets.1.norm2": "DownBlocks.Chain_12.ResidualBlock.Chain.GroupNorm_2",
|
||||
"mid_block.resnets.0.norm1": "MiddleBlock.ResidualBlock_1.Chain.GroupNorm_1",
|
||||
"mid_block.resnets.0.norm2": "MiddleBlock.ResidualBlock_1.Chain.GroupNorm_2",
|
||||
"mid_block.attentions.0.norm": "MiddleBlock.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"mid_block.resnets.1.norm1": "MiddleBlock.ResidualBlock_2.Chain.GroupNorm_1",
|
||||
"mid_block.resnets.1.norm2": "MiddleBlock.ResidualBlock_2.Chain.GroupNorm_2",
|
||||
"down_blocks.2.resnets.0.conv2": "DownBlocks.Chain_8.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.2.resnets.1.conv1": "DownBlocks.Chain_9.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.2.resnets.1.conv2": "DownBlocks.Chain_9.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.2.downsamplers.0.conv": "DownBlocks.Chain_10.Downsample.Conv2d",
|
||||
"down_blocks.3.resnets.0.conv1": "DownBlocks.Chain_11.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.3.resnets.0.conv2": "DownBlocks.Chain_11.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.3.resnets.1.conv1": "DownBlocks.Chain_12.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.3.resnets.1.conv2": "DownBlocks.Chain_12.ResidualBlock.Chain.Conv2d",
|
||||
"mid_block.resnets.0.conv1": "MiddleBlock.ResidualBlock_1.Chain.RangeAdapter2d.Conv2d",
|
||||
"mid_block.resnets.0.conv2": "MiddleBlock.ResidualBlock_1.Chain.Conv2d",
|
||||
"mid_block.resnets.1.conv1": "MiddleBlock.ResidualBlock_2.Chain.RangeAdapter2d.Conv2d",
|
||||
"mid_block.resnets.1.conv2": "MiddleBlock.ResidualBlock_2.Chain.Conv2d",
|
||||
"down_blocks.2.resnets.0.conv_shortcut": "DownBlocks.Chain_8.ResidualBlock.Conv2d",
|
||||
"down_blocks.2.attentions.0.proj_in": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.2.attentions.0.proj_out": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_down_blocks.7": "DownBlocks.Chain_8.Passthrough.Conv2d",
|
||||
"down_blocks.2.attentions.1.proj_in": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.2.attentions.1.proj_out": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_down_blocks.8": "DownBlocks.Chain_9.Passthrough.Conv2d",
|
||||
"controlnet_down_blocks.9": "DownBlocks.Chain_10.Passthrough.Conv2d",
|
||||
"controlnet_down_blocks.10": "DownBlocks.Chain_11.Passthrough.Conv2d",
|
||||
"controlnet_down_blocks.11": "DownBlocks.Chain_12.Passthrough.Conv2d",
|
||||
"mid_block.attentions.0.proj_in": "MiddleBlock.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"mid_block.attentions.0.proj_out": "MiddleBlock.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"controlnet_mid_block": "MiddleBlock.Passthrough.Conv2d",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.norm1": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.norm2": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.norm3": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.norm1": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.norm2": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.norm3": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"mid_block.attentions.0.transformer_blocks.0.norm1": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"mid_block.attentions.0.transformer_blocks.0.norm2": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"mid_block.attentions.0.transformer_blocks.0.norm3": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_q": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_k": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_v": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_q": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_k": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_v": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"mid_block.attentions.0.transformer_blocks.0.ff.net.2": "MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
@ -0,0 +1,44 @@
|
||||
{
|
||||
"name_map": {
|
||||
"image_proj.proj.weight": "image_proj.Linear.weight",
|
||||
"image_proj.proj.bias": "image_proj.Linear.bias",
|
||||
"image_proj.norm.weight": "image_proj.LayerNorm.weight",
|
||||
"image_proj.norm.bias": "image_proj.LayerNorm.bias",
|
||||
"ip_adapter.1.to_k_ip.weight": "ip_adapter.000.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.1.to_v_ip.weight": "ip_adapter.000.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_k_ip.weight": "ip_adapter.001.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_v_ip.weight": "ip_adapter.001.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_k_ip.weight": "ip_adapter.002.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_v_ip.weight": "ip_adapter.002.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_k_ip.weight": "ip_adapter.003.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_v_ip.weight": "ip_adapter.003.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_k_ip.weight": "ip_adapter.004.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_v_ip.weight": "ip_adapter.004.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_k_ip.weight": "ip_adapter.005.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_v_ip.weight": "ip_adapter.005.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_k_ip.weight": "ip_adapter.006.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_v_ip.weight": "ip_adapter.006.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_k_ip.weight": "ip_adapter.007.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_v_ip.weight": "ip_adapter.007.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_k_ip.weight": "ip_adapter.008.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_v_ip.weight": "ip_adapter.008.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_k_ip.weight": "ip_adapter.009.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_v_ip.weight": "ip_adapter.009.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_k_ip.weight": "ip_adapter.010.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_v_ip.weight": "ip_adapter.010.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_k_ip.weight": "ip_adapter.011.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_v_ip.weight": "ip_adapter.011.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_k_ip.weight": "ip_adapter.012.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_v_ip.weight": "ip_adapter.012.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_k_ip.weight": "ip_adapter.013.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_v_ip.weight": "ip_adapter.013.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_k_ip.weight": "ip_adapter.014.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_v_ip.weight": "ip_adapter.014.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_k_ip.weight": "ip_adapter.015.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_v_ip.weight": "ip_adapter.015.Distribute.Parallel_2.Chain_2.Linear.weight"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
@ -0,0 +1,152 @@
|
||||
{
|
||||
"name_map": {
|
||||
"image_proj.proj.weight": "image_proj.Linear.weight",
|
||||
"image_proj.proj.bias": "image_proj.Linear.bias",
|
||||
"image_proj.norm.weight": "image_proj.LayerNorm.weight",
|
||||
"image_proj.norm.bias": "image_proj.LayerNorm.bias",
|
||||
"ip_adapter.1.to_k_ip.weight": "ip_adapter.000.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.1.to_v_ip.weight": "ip_adapter.000.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_k_ip.weight": "ip_adapter.001.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_v_ip.weight": "ip_adapter.001.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_k_ip.weight": "ip_adapter.002.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_v_ip.weight": "ip_adapter.002.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_k_ip.weight": "ip_adapter.003.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_v_ip.weight": "ip_adapter.003.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_k_ip.weight": "ip_adapter.004.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_v_ip.weight": "ip_adapter.004.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_k_ip.weight": "ip_adapter.005.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_v_ip.weight": "ip_adapter.005.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_k_ip.weight": "ip_adapter.006.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_v_ip.weight": "ip_adapter.006.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_k_ip.weight": "ip_adapter.007.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_v_ip.weight": "ip_adapter.007.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_k_ip.weight": "ip_adapter.008.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_v_ip.weight": "ip_adapter.008.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_k_ip.weight": "ip_adapter.009.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_v_ip.weight": "ip_adapter.009.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_k_ip.weight": "ip_adapter.010.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_v_ip.weight": "ip_adapter.010.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_k_ip.weight": "ip_adapter.011.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_v_ip.weight": "ip_adapter.011.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_k_ip.weight": "ip_adapter.012.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_v_ip.weight": "ip_adapter.012.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_k_ip.weight": "ip_adapter.013.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_v_ip.weight": "ip_adapter.013.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_k_ip.weight": "ip_adapter.014.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_v_ip.weight": "ip_adapter.014.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_k_ip.weight": "ip_adapter.015.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_v_ip.weight": "ip_adapter.015.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.33.to_k_ip.weight": "ip_adapter.016.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.33.to_v_ip.weight": "ip_adapter.016.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.35.to_k_ip.weight": "ip_adapter.017.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.35.to_v_ip.weight": "ip_adapter.017.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.37.to_k_ip.weight": "ip_adapter.018.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.37.to_v_ip.weight": "ip_adapter.018.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.39.to_k_ip.weight": "ip_adapter.019.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.39.to_v_ip.weight": "ip_adapter.019.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.41.to_k_ip.weight": "ip_adapter.020.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.41.to_v_ip.weight": "ip_adapter.020.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.43.to_k_ip.weight": "ip_adapter.021.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.43.to_v_ip.weight": "ip_adapter.021.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.45.to_k_ip.weight": "ip_adapter.022.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.45.to_v_ip.weight": "ip_adapter.022.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.47.to_k_ip.weight": "ip_adapter.023.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.47.to_v_ip.weight": "ip_adapter.023.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.121.to_k_ip.weight": "ip_adapter.024.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.121.to_v_ip.weight": "ip_adapter.024.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.123.to_k_ip.weight": "ip_adapter.025.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.123.to_v_ip.weight": "ip_adapter.025.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.125.to_k_ip.weight": "ip_adapter.026.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.125.to_v_ip.weight": "ip_adapter.026.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.127.to_k_ip.weight": "ip_adapter.027.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.127.to_v_ip.weight": "ip_adapter.027.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.129.to_k_ip.weight": "ip_adapter.028.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.129.to_v_ip.weight": "ip_adapter.028.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.131.to_k_ip.weight": "ip_adapter.029.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.131.to_v_ip.weight": "ip_adapter.029.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.133.to_k_ip.weight": "ip_adapter.030.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.133.to_v_ip.weight": "ip_adapter.030.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.135.to_k_ip.weight": "ip_adapter.031.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.135.to_v_ip.weight": "ip_adapter.031.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.137.to_k_ip.weight": "ip_adapter.032.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.137.to_v_ip.weight": "ip_adapter.032.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.139.to_k_ip.weight": "ip_adapter.033.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.139.to_v_ip.weight": "ip_adapter.033.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.49.to_k_ip.weight": "ip_adapter.034.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.49.to_v_ip.weight": "ip_adapter.034.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.51.to_k_ip.weight": "ip_adapter.035.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.51.to_v_ip.weight": "ip_adapter.035.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.53.to_k_ip.weight": "ip_adapter.036.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.53.to_v_ip.weight": "ip_adapter.036.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.55.to_k_ip.weight": "ip_adapter.037.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.55.to_v_ip.weight": "ip_adapter.037.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.57.to_k_ip.weight": "ip_adapter.038.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.57.to_v_ip.weight": "ip_adapter.038.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.59.to_k_ip.weight": "ip_adapter.039.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.59.to_v_ip.weight": "ip_adapter.039.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.61.to_k_ip.weight": "ip_adapter.040.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.61.to_v_ip.weight": "ip_adapter.040.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.63.to_k_ip.weight": "ip_adapter.041.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.63.to_v_ip.weight": "ip_adapter.041.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.65.to_k_ip.weight": "ip_adapter.042.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.65.to_v_ip.weight": "ip_adapter.042.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.67.to_k_ip.weight": "ip_adapter.043.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.67.to_v_ip.weight": "ip_adapter.043.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.69.to_k_ip.weight": "ip_adapter.044.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.69.to_v_ip.weight": "ip_adapter.044.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.71.to_k_ip.weight": "ip_adapter.045.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.71.to_v_ip.weight": "ip_adapter.045.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.73.to_k_ip.weight": "ip_adapter.046.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.73.to_v_ip.weight": "ip_adapter.046.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.75.to_k_ip.weight": "ip_adapter.047.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.75.to_v_ip.weight": "ip_adapter.047.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.77.to_k_ip.weight": "ip_adapter.048.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.77.to_v_ip.weight": "ip_adapter.048.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.79.to_k_ip.weight": "ip_adapter.049.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.79.to_v_ip.weight": "ip_adapter.049.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.81.to_k_ip.weight": "ip_adapter.050.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.81.to_v_ip.weight": "ip_adapter.050.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.83.to_k_ip.weight": "ip_adapter.051.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.83.to_v_ip.weight": "ip_adapter.051.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.85.to_k_ip.weight": "ip_adapter.052.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.85.to_v_ip.weight": "ip_adapter.052.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.87.to_k_ip.weight": "ip_adapter.053.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.87.to_v_ip.weight": "ip_adapter.053.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.89.to_k_ip.weight": "ip_adapter.054.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.89.to_v_ip.weight": "ip_adapter.054.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.91.to_k_ip.weight": "ip_adapter.055.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.91.to_v_ip.weight": "ip_adapter.055.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.93.to_k_ip.weight": "ip_adapter.056.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.93.to_v_ip.weight": "ip_adapter.056.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.95.to_k_ip.weight": "ip_adapter.057.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.95.to_v_ip.weight": "ip_adapter.057.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.97.to_k_ip.weight": "ip_adapter.058.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.97.to_v_ip.weight": "ip_adapter.058.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.99.to_k_ip.weight": "ip_adapter.059.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.99.to_v_ip.weight": "ip_adapter.059.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.101.to_k_ip.weight": "ip_adapter.060.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.101.to_v_ip.weight": "ip_adapter.060.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.103.to_k_ip.weight": "ip_adapter.061.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.103.to_v_ip.weight": "ip_adapter.061.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.105.to_k_ip.weight": "ip_adapter.062.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.105.to_v_ip.weight": "ip_adapter.062.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.107.to_k_ip.weight": "ip_adapter.063.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.107.to_v_ip.weight": "ip_adapter.063.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.109.to_k_ip.weight": "ip_adapter.064.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.109.to_v_ip.weight": "ip_adapter.064.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.111.to_k_ip.weight": "ip_adapter.065.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.111.to_v_ip.weight": "ip_adapter.065.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.113.to_k_ip.weight": "ip_adapter.066.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.113.to_v_ip.weight": "ip_adapter.066.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.115.to_k_ip.weight": "ip_adapter.067.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.115.to_v_ip.weight": "ip_adapter.067.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.117.to_k_ip.weight": "ip_adapter.068.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.117.to_v_ip.weight": "ip_adapter.068.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.119.to_k_ip.weight": "ip_adapter.069.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.119.to_v_ip.weight": "ip_adapter.069.Distribute.Parallel_2.Chain_2.Linear.weight"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
@ -0,0 +1,96 @@
|
||||
{
|
||||
"name_map": {
|
||||
"image_proj.latents": "image_proj.LatentsToken.Parameter.weight",
|
||||
"image_proj.proj_in.weight": "image_proj.Linear_1.weight",
|
||||
"image_proj.proj_in.bias": "image_proj.Linear_1.bias",
|
||||
"image_proj.proj_out.weight": "image_proj.Linear_2.weight",
|
||||
"image_proj.proj_out.bias": "image_proj.Linear_2.bias",
|
||||
"image_proj.norm_out.weight": "image_proj.LayerNorm.weight",
|
||||
"image_proj.norm_out.bias": "image_proj.LayerNorm.bias",
|
||||
"image_proj.layers.0.0.norm1.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.0.0.norm1.bias": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.0.0.norm2.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.0.0.norm2.bias": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.0.0.to_q.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.0.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.0.0.to_out.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.0.1.0.weight": "image_proj.Transformer.TransformerLayer_1.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.0.1.0.bias": "image_proj.Transformer.TransformerLayer_1.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.0.1.1.weight": "image_proj.Transformer.TransformerLayer_1.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.0.1.3.weight": "image_proj.Transformer.TransformerLayer_1.Residual_2.FeedForward.Linear_2.weight",
|
||||
"image_proj.layers.1.0.norm1.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.1.0.norm1.bias": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.1.0.norm2.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.1.0.norm2.bias": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.1.0.to_q.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.1.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.1.0.to_out.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.1.1.0.weight": "image_proj.Transformer.TransformerLayer_2.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.1.1.0.bias": "image_proj.Transformer.TransformerLayer_2.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.1.1.1.weight": "image_proj.Transformer.TransformerLayer_2.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.1.1.3.weight": "image_proj.Transformer.TransformerLayer_2.Residual_2.FeedForward.Linear_2.weight",
|
||||
"image_proj.layers.2.0.norm1.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.2.0.norm1.bias": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.2.0.norm2.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.2.0.norm2.bias": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.2.0.to_q.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.2.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.2.0.to_out.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.2.1.0.weight": "image_proj.Transformer.TransformerLayer_3.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.2.1.0.bias": "image_proj.Transformer.TransformerLayer_3.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.2.1.1.weight": "image_proj.Transformer.TransformerLayer_3.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.2.1.3.weight": "image_proj.Transformer.TransformerLayer_3.Residual_2.FeedForward.Linear_2.weight",
|
||||
"image_proj.layers.3.0.norm1.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.3.0.norm1.bias": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.3.0.norm2.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.3.0.norm2.bias": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.3.0.to_q.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.3.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.3.0.to_out.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.3.1.0.weight": "image_proj.Transformer.TransformerLayer_4.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.3.1.0.bias": "image_proj.Transformer.TransformerLayer_4.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.3.1.1.weight": "image_proj.Transformer.TransformerLayer_4.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.3.1.3.weight": "image_proj.Transformer.TransformerLayer_4.Residual_2.FeedForward.Linear_2.weight",
|
||||
"ip_adapter.1.to_k_ip.weight": "ip_adapter.000.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.1.to_v_ip.weight": "ip_adapter.000.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_k_ip.weight": "ip_adapter.001.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_v_ip.weight": "ip_adapter.001.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_k_ip.weight": "ip_adapter.002.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_v_ip.weight": "ip_adapter.002.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_k_ip.weight": "ip_adapter.003.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_v_ip.weight": "ip_adapter.003.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_k_ip.weight": "ip_adapter.004.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_v_ip.weight": "ip_adapter.004.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_k_ip.weight": "ip_adapter.005.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_v_ip.weight": "ip_adapter.005.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_k_ip.weight": "ip_adapter.006.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_v_ip.weight": "ip_adapter.006.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_k_ip.weight": "ip_adapter.007.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_v_ip.weight": "ip_adapter.007.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_k_ip.weight": "ip_adapter.008.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_v_ip.weight": "ip_adapter.008.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_k_ip.weight": "ip_adapter.009.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_v_ip.weight": "ip_adapter.009.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_k_ip.weight": "ip_adapter.010.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_v_ip.weight": "ip_adapter.010.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_k_ip.weight": "ip_adapter.011.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_v_ip.weight": "ip_adapter.011.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_k_ip.weight": "ip_adapter.012.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_v_ip.weight": "ip_adapter.012.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_k_ip.weight": "ip_adapter.013.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_v_ip.weight": "ip_adapter.013.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_k_ip.weight": "ip_adapter.014.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_v_ip.weight": "ip_adapter.014.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_k_ip.weight": "ip_adapter.015.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_v_ip.weight": "ip_adapter.015.Distribute.Parallel_2.Chain_2.Linear.weight"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {
|
||||
"image_proj.LatentsToken.Parameter.weight": [
|
||||
16,
|
||||
768
|
||||
]
|
||||
}
|
||||
}
|
@ -0,0 +1,204 @@
|
||||
{
|
||||
"name_map": {
|
||||
"image_proj.latents": "image_proj.LatentsToken.Parameter.weight",
|
||||
"image_proj.proj_in.weight": "image_proj.Linear_1.weight",
|
||||
"image_proj.proj_in.bias": "image_proj.Linear_1.bias",
|
||||
"image_proj.proj_out.weight": "image_proj.Linear_2.weight",
|
||||
"image_proj.proj_out.bias": "image_proj.Linear_2.bias",
|
||||
"image_proj.norm_out.weight": "image_proj.LayerNorm.weight",
|
||||
"image_proj.norm_out.bias": "image_proj.LayerNorm.bias",
|
||||
"image_proj.layers.0.0.norm1.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.0.0.norm1.bias": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.0.0.norm2.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.0.0.norm2.bias": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.0.0.to_q.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.0.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.0.0.to_out.weight": "image_proj.Transformer.TransformerLayer_1.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.0.1.0.weight": "image_proj.Transformer.TransformerLayer_1.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.0.1.0.bias": "image_proj.Transformer.TransformerLayer_1.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.0.1.1.weight": "image_proj.Transformer.TransformerLayer_1.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.0.1.3.weight": "image_proj.Transformer.TransformerLayer_1.Residual_2.FeedForward.Linear_2.weight",
|
||||
"image_proj.layers.1.0.norm1.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.1.0.norm1.bias": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.1.0.norm2.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.1.0.norm2.bias": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.1.0.to_q.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.1.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.1.0.to_out.weight": "image_proj.Transformer.TransformerLayer_2.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.1.1.0.weight": "image_proj.Transformer.TransformerLayer_2.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.1.1.0.bias": "image_proj.Transformer.TransformerLayer_2.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.1.1.1.weight": "image_proj.Transformer.TransformerLayer_2.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.1.1.3.weight": "image_proj.Transformer.TransformerLayer_2.Residual_2.FeedForward.Linear_2.weight",
|
||||
"image_proj.layers.2.0.norm1.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.2.0.norm1.bias": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.2.0.norm2.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.2.0.norm2.bias": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.2.0.to_q.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.2.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.2.0.to_out.weight": "image_proj.Transformer.TransformerLayer_3.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.2.1.0.weight": "image_proj.Transformer.TransformerLayer_3.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.2.1.0.bias": "image_proj.Transformer.TransformerLayer_3.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.2.1.1.weight": "image_proj.Transformer.TransformerLayer_3.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.2.1.3.weight": "image_proj.Transformer.TransformerLayer_3.Residual_2.FeedForward.Linear_2.weight",
|
||||
"image_proj.layers.3.0.norm1.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.weight",
|
||||
"image_proj.layers.3.0.norm1.bias": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_1.bias",
|
||||
"image_proj.layers.3.0.norm2.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.weight",
|
||||
"image_proj.layers.3.0.norm2.bias": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Distribute.LayerNorm_2.bias",
|
||||
"image_proj.layers.3.0.to_q.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Parallel.Chain_2.Linear.weight",
|
||||
"image_proj.layers.3.0.to_kv.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Parallel.Chain_1.Linear.weight",
|
||||
"image_proj.layers.3.0.to_out.weight": "image_proj.Transformer.TransformerLayer_4.Residual_1.PerceiverAttention.Linear.weight",
|
||||
"image_proj.layers.3.1.0.weight": "image_proj.Transformer.TransformerLayer_4.Residual_2.LayerNorm.weight",
|
||||
"image_proj.layers.3.1.0.bias": "image_proj.Transformer.TransformerLayer_4.Residual_2.LayerNorm.bias",
|
||||
"image_proj.layers.3.1.1.weight": "image_proj.Transformer.TransformerLayer_4.Residual_2.FeedForward.Linear_1.weight",
|
||||
"image_proj.layers.3.1.3.weight": "image_proj.Transformer.TransformerLayer_4.Residual_2.FeedForward.Linear_2.weight",
|
||||
"ip_adapter.1.to_k_ip.weight": "ip_adapter.000.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.1.to_v_ip.weight": "ip_adapter.000.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_k_ip.weight": "ip_adapter.001.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.3.to_v_ip.weight": "ip_adapter.001.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_k_ip.weight": "ip_adapter.002.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.5.to_v_ip.weight": "ip_adapter.002.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_k_ip.weight": "ip_adapter.003.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.7.to_v_ip.weight": "ip_adapter.003.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_k_ip.weight": "ip_adapter.004.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.9.to_v_ip.weight": "ip_adapter.004.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_k_ip.weight": "ip_adapter.005.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.11.to_v_ip.weight": "ip_adapter.005.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_k_ip.weight": "ip_adapter.006.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.13.to_v_ip.weight": "ip_adapter.006.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_k_ip.weight": "ip_adapter.007.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.15.to_v_ip.weight": "ip_adapter.007.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_k_ip.weight": "ip_adapter.008.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.17.to_v_ip.weight": "ip_adapter.008.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_k_ip.weight": "ip_adapter.009.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.19.to_v_ip.weight": "ip_adapter.009.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_k_ip.weight": "ip_adapter.010.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.21.to_v_ip.weight": "ip_adapter.010.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_k_ip.weight": "ip_adapter.011.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.23.to_v_ip.weight": "ip_adapter.011.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_k_ip.weight": "ip_adapter.012.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.25.to_v_ip.weight": "ip_adapter.012.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_k_ip.weight": "ip_adapter.013.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.27.to_v_ip.weight": "ip_adapter.013.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_k_ip.weight": "ip_adapter.014.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.29.to_v_ip.weight": "ip_adapter.014.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_k_ip.weight": "ip_adapter.015.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.31.to_v_ip.weight": "ip_adapter.015.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.33.to_k_ip.weight": "ip_adapter.016.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.33.to_v_ip.weight": "ip_adapter.016.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.35.to_k_ip.weight": "ip_adapter.017.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.35.to_v_ip.weight": "ip_adapter.017.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.37.to_k_ip.weight": "ip_adapter.018.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.37.to_v_ip.weight": "ip_adapter.018.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.39.to_k_ip.weight": "ip_adapter.019.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.39.to_v_ip.weight": "ip_adapter.019.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.41.to_k_ip.weight": "ip_adapter.020.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.41.to_v_ip.weight": "ip_adapter.020.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.43.to_k_ip.weight": "ip_adapter.021.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.43.to_v_ip.weight": "ip_adapter.021.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.45.to_k_ip.weight": "ip_adapter.022.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.45.to_v_ip.weight": "ip_adapter.022.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.47.to_k_ip.weight": "ip_adapter.023.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.47.to_v_ip.weight": "ip_adapter.023.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.121.to_k_ip.weight": "ip_adapter.024.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.121.to_v_ip.weight": "ip_adapter.024.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.123.to_k_ip.weight": "ip_adapter.025.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.123.to_v_ip.weight": "ip_adapter.025.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.125.to_k_ip.weight": "ip_adapter.026.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.125.to_v_ip.weight": "ip_adapter.026.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.127.to_k_ip.weight": "ip_adapter.027.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.127.to_v_ip.weight": "ip_adapter.027.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.129.to_k_ip.weight": "ip_adapter.028.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.129.to_v_ip.weight": "ip_adapter.028.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.131.to_k_ip.weight": "ip_adapter.029.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.131.to_v_ip.weight": "ip_adapter.029.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.133.to_k_ip.weight": "ip_adapter.030.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.133.to_v_ip.weight": "ip_adapter.030.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.135.to_k_ip.weight": "ip_adapter.031.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.135.to_v_ip.weight": "ip_adapter.031.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.137.to_k_ip.weight": "ip_adapter.032.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.137.to_v_ip.weight": "ip_adapter.032.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.139.to_k_ip.weight": "ip_adapter.033.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.139.to_v_ip.weight": "ip_adapter.033.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.49.to_k_ip.weight": "ip_adapter.034.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.49.to_v_ip.weight": "ip_adapter.034.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.51.to_k_ip.weight": "ip_adapter.035.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.51.to_v_ip.weight": "ip_adapter.035.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.53.to_k_ip.weight": "ip_adapter.036.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.53.to_v_ip.weight": "ip_adapter.036.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.55.to_k_ip.weight": "ip_adapter.037.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.55.to_v_ip.weight": "ip_adapter.037.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.57.to_k_ip.weight": "ip_adapter.038.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.57.to_v_ip.weight": "ip_adapter.038.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.59.to_k_ip.weight": "ip_adapter.039.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.59.to_v_ip.weight": "ip_adapter.039.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.61.to_k_ip.weight": "ip_adapter.040.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.61.to_v_ip.weight": "ip_adapter.040.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.63.to_k_ip.weight": "ip_adapter.041.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.63.to_v_ip.weight": "ip_adapter.041.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.65.to_k_ip.weight": "ip_adapter.042.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.65.to_v_ip.weight": "ip_adapter.042.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.67.to_k_ip.weight": "ip_adapter.043.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.67.to_v_ip.weight": "ip_adapter.043.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.69.to_k_ip.weight": "ip_adapter.044.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.69.to_v_ip.weight": "ip_adapter.044.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.71.to_k_ip.weight": "ip_adapter.045.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.71.to_v_ip.weight": "ip_adapter.045.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.73.to_k_ip.weight": "ip_adapter.046.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.73.to_v_ip.weight": "ip_adapter.046.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.75.to_k_ip.weight": "ip_adapter.047.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.75.to_v_ip.weight": "ip_adapter.047.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.77.to_k_ip.weight": "ip_adapter.048.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.77.to_v_ip.weight": "ip_adapter.048.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.79.to_k_ip.weight": "ip_adapter.049.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.79.to_v_ip.weight": "ip_adapter.049.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.81.to_k_ip.weight": "ip_adapter.050.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.81.to_v_ip.weight": "ip_adapter.050.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.83.to_k_ip.weight": "ip_adapter.051.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.83.to_v_ip.weight": "ip_adapter.051.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.85.to_k_ip.weight": "ip_adapter.052.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.85.to_v_ip.weight": "ip_adapter.052.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.87.to_k_ip.weight": "ip_adapter.053.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.87.to_v_ip.weight": "ip_adapter.053.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.89.to_k_ip.weight": "ip_adapter.054.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.89.to_v_ip.weight": "ip_adapter.054.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.91.to_k_ip.weight": "ip_adapter.055.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.91.to_v_ip.weight": "ip_adapter.055.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.93.to_k_ip.weight": "ip_adapter.056.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.93.to_v_ip.weight": "ip_adapter.056.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.95.to_k_ip.weight": "ip_adapter.057.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.95.to_v_ip.weight": "ip_adapter.057.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.97.to_k_ip.weight": "ip_adapter.058.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.97.to_v_ip.weight": "ip_adapter.058.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.99.to_k_ip.weight": "ip_adapter.059.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.99.to_v_ip.weight": "ip_adapter.059.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.101.to_k_ip.weight": "ip_adapter.060.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.101.to_v_ip.weight": "ip_adapter.060.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.103.to_k_ip.weight": "ip_adapter.061.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.103.to_v_ip.weight": "ip_adapter.061.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.105.to_k_ip.weight": "ip_adapter.062.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.105.to_v_ip.weight": "ip_adapter.062.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.107.to_k_ip.weight": "ip_adapter.063.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.107.to_v_ip.weight": "ip_adapter.063.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.109.to_k_ip.weight": "ip_adapter.064.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.109.to_v_ip.weight": "ip_adapter.064.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.111.to_k_ip.weight": "ip_adapter.065.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.111.to_v_ip.weight": "ip_adapter.065.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.113.to_k_ip.weight": "ip_adapter.066.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.113.to_v_ip.weight": "ip_adapter.066.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.115.to_k_ip.weight": "ip_adapter.067.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.115.to_v_ip.weight": "ip_adapter.067.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.117.to_k_ip.weight": "ip_adapter.068.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.117.to_v_ip.weight": "ip_adapter.068.Distribute.Parallel_2.Chain_2.Linear.weight",
|
||||
"ip_adapter.119.to_k_ip.weight": "ip_adapter.069.Distribute.Parallel_1.Chain_2.Linear.weight",
|
||||
"ip_adapter.119.to_v_ip.weight": "ip_adapter.069.Distribute.Parallel_2.Chain_2.Linear.weight"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {
|
||||
"image_proj.LatentsToken.Parameter.weight": [
|
||||
16,
|
||||
1280
|
||||
]
|
||||
}
|
||||
}
|
@ -0,0 +1,30 @@
|
||||
{
|
||||
"name_map": {
|
||||
"adapter.conv_in": "Conv2d",
|
||||
"adapter.body.0.resnets.0.block1": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.0.resnets.1.block1": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.0.resnets.0.block2": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.0.resnets.1.block2": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2",
|
||||
"adapter.body.1.downsample": "StatefulResidualBlocks_2.ResidualBlocks.Downsample2d",
|
||||
"adapter.body.2.downsample": "StatefulResidualBlocks_3.ResidualBlocks.Downsample2d",
|
||||
"adapter.body.3.downsample": "StatefulResidualBlocks_4.ResidualBlocks.Downsample2d",
|
||||
"adapter.body.1.in_conv": "StatefulResidualBlocks_2.ResidualBlocks.Conv2d",
|
||||
"adapter.body.1.resnets.0.block1": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.1.resnets.1.block1": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.1.resnets.0.block2": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.1.resnets.1.block2": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2",
|
||||
"adapter.body.2.in_conv": "StatefulResidualBlocks_3.ResidualBlocks.Conv2d",
|
||||
"adapter.body.2.resnets.0.block1": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.2.resnets.1.block1": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.3.resnets.0.block1": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.3.resnets.1.block1": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.2.resnets.0.block2": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.2.resnets.1.block2": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2",
|
||||
"adapter.body.3.resnets.0.block2": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.3.resnets.1.block2": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
@ -0,0 +1,28 @@
|
||||
{
|
||||
"name_map": {
|
||||
"adapter.conv_in": "Conv2d",
|
||||
"adapter.body.0.resnets.0.block1": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.0.resnets.1.block1": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.0.resnets.0.block2": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.0.resnets.1.block2": "StatefulResidualBlocks_1.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2",
|
||||
"adapter.body.1.in_conv": "StatefulResidualBlocks_2.ResidualBlocks.Conv2d",
|
||||
"adapter.body.1.resnets.0.block1": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.1.resnets.1.block1": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.1.resnets.0.block2": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.1.resnets.1.block2": "StatefulResidualBlocks_2.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2",
|
||||
"adapter.body.2.downsample": "StatefulResidualBlocks_3.ResidualBlocks.Downsample2d",
|
||||
"adapter.body.2.in_conv": "StatefulResidualBlocks_3.ResidualBlocks.Conv2d",
|
||||
"adapter.body.2.resnets.0.block1": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.2.resnets.1.block1": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.3.resnets.0.block1": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_1",
|
||||
"adapter.body.3.resnets.1.block1": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_1",
|
||||
"adapter.body.2.resnets.0.block2": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.2.resnets.1.block2": "StatefulResidualBlocks_3.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2",
|
||||
"adapter.body.3.resnets.0.block2": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_1.Conv2d_2",
|
||||
"adapter.body.3.resnets.1.block2": "StatefulResidualBlocks_4.ResidualBlocks.Chain.ResidualBlock_2.Conv2d_2"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
@ -0,0 +1,399 @@
|
||||
{
|
||||
"name_map": {
|
||||
"time_embedding.linear_1": "TimestepEncoder.RangeEncoder.Linear_1",
|
||||
"time_embedding.linear_2": "TimestepEncoder.RangeEncoder.Linear_2",
|
||||
"down_blocks.2.resnets.0.time_emb_proj": "DownBlocks.Chain_8.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.2.resnets.1.time_emb_proj": "DownBlocks.Chain_9.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.3.resnets.0.time_emb_proj": "DownBlocks.Chain_11.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.3.resnets.1.time_emb_proj": "DownBlocks.Chain_12.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"mid_block.resnets.0.time_emb_proj": "Sum.MiddleBlock.ResidualBlock_1.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_out.0": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_out.0": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"mid_block.resnets.1.time_emb_proj": "Sum.MiddleBlock.ResidualBlock_2.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.0.resnets.0.time_emb_proj": "UpBlocks.Chain_1.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.0.resnets.1.time_emb_proj": "UpBlocks.Chain_2.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.0.resnets.2.time_emb_proj": "UpBlocks.Chain_3.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.1.resnets.0.time_emb_proj": "UpBlocks.Chain_4.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.1.resnets.1.time_emb_proj": "UpBlocks.Chain_5.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.1.resnets.2.time_emb_proj": "UpBlocks.Chain_6.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"conv_in": "DownBlocks.Chain_1.Conv2d",
|
||||
"down_blocks.0.resnets.0.norm1": "DownBlocks.Chain_2.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.0.resnets.0.norm2": "DownBlocks.Chain_2.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.0.attentions.0.norm": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.0.resnets.1.norm1": "DownBlocks.Chain_3.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.0.resnets.1.norm2": "DownBlocks.Chain_3.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.0.attentions.1.norm": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.1.resnets.0.norm1": "DownBlocks.Chain_5.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.3.resnets.0.norm2": "UpBlocks.Chain_10.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.3.attentions.0.norm": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.3.resnets.1.norm2": "UpBlocks.Chain_11.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.3.attentions.1.norm": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.3.resnets.2.norm2": "UpBlocks.Chain_12.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.3.attentions.2.norm": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"conv_norm_out": "Chain.GroupNorm",
|
||||
"down_blocks.0.resnets.0.conv1": "DownBlocks.Chain_2.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.0.resnets.0.conv2": "DownBlocks.Chain_2.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.0.resnets.1.conv1": "DownBlocks.Chain_3.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.0.resnets.1.conv2": "DownBlocks.Chain_3.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.0.downsamplers.0.conv": "DownBlocks.Chain_4.Downsample.Conv2d",
|
||||
"up_blocks.3.resnets.0.conv2": "UpBlocks.Chain_10.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.3.resnets.1.conv2": "UpBlocks.Chain_11.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.3.resnets.2.conv2": "UpBlocks.Chain_12.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.0.resnets.0.time_emb_proj": "DownBlocks.Chain_2.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.0.resnets.1.time_emb_proj": "DownBlocks.Chain_3.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.3.resnets.0.time_emb_proj": "UpBlocks.Chain_10.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.3.resnets.1.time_emb_proj": "UpBlocks.Chain_11.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.3.resnets.2.time_emb_proj": "UpBlocks.Chain_12.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.0.attentions.0.proj_in": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.0.attentions.0.proj_out": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"down_blocks.0.attentions.1.proj_in": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.0.attentions.1.proj_out": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.3.attentions.0.proj_in": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.3.attentions.0.proj_out": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.3.attentions.1.proj_in": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.3.attentions.1.proj_out": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.3.attentions.2.proj_in": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.3.attentions.2.proj_out": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.norm1": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.norm2": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.norm3": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.norm1": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.norm2": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.norm3": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.norm1": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.norm2": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.norm3": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.norm1": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.norm2": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.norm3": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.norm1": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.norm2": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.norm3": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.0.attentions.0.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_2.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.0.attentions.1.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_3.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.3.attentions.0.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_10.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.3.attentions.1.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_11.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.3.attentions.2.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_12.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.1.resnets.0.conv1": "DownBlocks.Chain_5.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.1.resnets.0.time_emb_proj": "DownBlocks.Chain_5.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.1.resnets.1.time_emb_proj": "DownBlocks.Chain_6.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.2.resnets.0.time_emb_proj": "UpBlocks.Chain_7.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.2.resnets.1.time_emb_proj": "UpBlocks.Chain_8.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"up_blocks.2.resnets.2.time_emb_proj": "UpBlocks.Chain_9.ResidualBlock.Chain.RangeAdapter2d.Chain.Linear",
|
||||
"down_blocks.1.resnets.0.norm2": "DownBlocks.Chain_5.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.1.attentions.0.norm": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.1.resnets.1.norm1": "DownBlocks.Chain_6.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.1.resnets.1.norm2": "DownBlocks.Chain_6.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.1.attentions.1.norm": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.2.resnets.0.norm1": "DownBlocks.Chain_8.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.2.resnets.0.norm2": "UpBlocks.Chain_7.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.2.attentions.0.norm": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.2.resnets.1.norm2": "UpBlocks.Chain_8.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.2.attentions.1.norm": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.2.resnets.2.norm2": "UpBlocks.Chain_9.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.2.attentions.2.norm": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.3.resnets.1.norm1": "UpBlocks.Chain_11.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.3.resnets.2.norm1": "UpBlocks.Chain_12.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.1.resnets.0.conv2": "DownBlocks.Chain_5.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.1.resnets.1.conv1": "DownBlocks.Chain_6.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.1.resnets.1.conv2": "DownBlocks.Chain_6.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.1.downsamplers.0.conv": "DownBlocks.Chain_7.Downsample.Conv2d",
|
||||
"up_blocks.2.resnets.0.conv2": "UpBlocks.Chain_7.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.2.resnets.1.conv2": "UpBlocks.Chain_8.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.2.resnets.2.conv2": "UpBlocks.Chain_9.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.2.upsamplers.0.conv": "UpBlocks.Chain_9.Upsample.Conv2d",
|
||||
"down_blocks.1.resnets.0.conv_shortcut": "DownBlocks.Chain_5.ResidualBlock.Conv2d",
|
||||
"down_blocks.1.attentions.0.proj_in": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.1.attentions.0.proj_out": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"down_blocks.1.attentions.1.proj_in": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.1.attentions.1.proj_out": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.2.attentions.0.proj_in": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.2.attentions.0.proj_out": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.2.attentions.1.proj_in": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.2.attentions.1.proj_out": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.2.attentions.2.proj_in": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.2.attentions.2.proj_out": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.norm1": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.norm2": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.norm3": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.norm1": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.norm2": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.norm3": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.norm1": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.norm2": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.norm3": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.norm1": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.norm2": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.norm3": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.norm1": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.norm2": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.norm3": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn1.to_out.0": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_out.0": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn1.to_out.0": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Linear",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_out.0": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Linear",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_7.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.2.attentions.2.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.2.resnets.0.conv1": "DownBlocks.Chain_8.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.2.resnets.0.norm2": "DownBlocks.Chain_8.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.2.attentions.0.norm": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.2.resnets.1.norm1": "DownBlocks.Chain_9.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.2.resnets.1.norm2": "DownBlocks.Chain_9.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.2.attentions.1.norm": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"down_blocks.3.resnets.0.norm1": "DownBlocks.Chain_11.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.3.resnets.0.norm2": "DownBlocks.Chain_11.ResidualBlock.Chain.GroupNorm_2",
|
||||
"down_blocks.3.resnets.1.norm1": "DownBlocks.Chain_12.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.3.resnets.1.norm2": "DownBlocks.Chain_12.ResidualBlock.Chain.GroupNorm_2",
|
||||
"mid_block.resnets.0.norm1": "Sum.MiddleBlock.ResidualBlock_1.Chain.GroupNorm_1",
|
||||
"mid_block.resnets.0.norm2": "Sum.MiddleBlock.ResidualBlock_1.Chain.GroupNorm_2",
|
||||
"mid_block.attentions.0.norm": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"mid_block.resnets.1.norm1": "Sum.MiddleBlock.ResidualBlock_2.Chain.GroupNorm_1",
|
||||
"mid_block.resnets.1.norm2": "Sum.MiddleBlock.ResidualBlock_2.Chain.GroupNorm_2",
|
||||
"up_blocks.0.resnets.0.norm2": "UpBlocks.Chain_1.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.0.resnets.1.norm2": "UpBlocks.Chain_2.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.0.resnets.2.norm2": "UpBlocks.Chain_3.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.1.resnets.0.norm2": "UpBlocks.Chain_4.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.1.attentions.0.norm": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.1.resnets.1.norm2": "UpBlocks.Chain_5.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.1.attentions.1.norm": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.1.resnets.2.norm2": "UpBlocks.Chain_6.ResidualBlock.Chain.GroupNorm_2",
|
||||
"up_blocks.1.attentions.2.norm": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_1.GroupNorm",
|
||||
"up_blocks.2.resnets.1.norm1": "UpBlocks.Chain_8.ResidualBlock.Chain.GroupNorm_1",
|
||||
"down_blocks.2.resnets.0.conv2": "DownBlocks.Chain_8.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.2.resnets.1.conv1": "DownBlocks.Chain_9.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.2.resnets.1.conv2": "DownBlocks.Chain_9.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.2.downsamplers.0.conv": "DownBlocks.Chain_10.Downsample.Conv2d",
|
||||
"down_blocks.3.resnets.0.conv1": "DownBlocks.Chain_11.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.3.resnets.0.conv2": "DownBlocks.Chain_11.ResidualBlock.Chain.Conv2d",
|
||||
"down_blocks.3.resnets.1.conv1": "DownBlocks.Chain_12.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"down_blocks.3.resnets.1.conv2": "DownBlocks.Chain_12.ResidualBlock.Chain.Conv2d",
|
||||
"mid_block.resnets.0.conv1": "Sum.MiddleBlock.ResidualBlock_1.Chain.RangeAdapter2d.Conv2d",
|
||||
"mid_block.resnets.0.conv2": "Sum.MiddleBlock.ResidualBlock_1.Chain.Conv2d",
|
||||
"mid_block.resnets.1.conv1": "Sum.MiddleBlock.ResidualBlock_2.Chain.RangeAdapter2d.Conv2d",
|
||||
"mid_block.resnets.1.conv2": "Sum.MiddleBlock.ResidualBlock_2.Chain.Conv2d",
|
||||
"up_blocks.0.resnets.0.conv2": "UpBlocks.Chain_1.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.0.resnets.1.conv2": "UpBlocks.Chain_2.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.0.resnets.2.conv2": "UpBlocks.Chain_3.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.0.upsamplers.0.conv": "UpBlocks.Chain_3.Upsample.Conv2d",
|
||||
"up_blocks.1.resnets.0.conv2": "UpBlocks.Chain_4.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.1.resnets.1.conv2": "UpBlocks.Chain_5.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.1.resnets.2.conv2": "UpBlocks.Chain_6.ResidualBlock.Chain.Conv2d",
|
||||
"up_blocks.1.upsamplers.0.conv": "UpBlocks.Chain_6.Upsample.Conv2d",
|
||||
"down_blocks.2.resnets.0.conv_shortcut": "DownBlocks.Chain_8.ResidualBlock.Conv2d",
|
||||
"down_blocks.2.attentions.0.proj_in": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.2.attentions.0.proj_out": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"down_blocks.2.attentions.1.proj_in": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"down_blocks.2.attentions.1.proj_out": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"mid_block.attentions.0.proj_in": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"mid_block.attentions.0.proj_out": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.1.attentions.0.proj_in": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.1.attentions.0.proj_out": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.1.attentions.1.proj_in": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.1.attentions.1.proj_out": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"up_blocks.1.attentions.2.proj_in": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_1.Conv2d",
|
||||
"up_blocks.1.attentions.2.proj_out": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_3.Conv2d",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.norm1": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.norm2": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.norm3": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.norm1": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.norm2": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.norm3": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"mid_block.attentions.0.transformer_blocks.0.norm1": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"mid_block.attentions.0.transformer_blocks.0.norm2": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"mid_block.attentions.0.transformer_blocks.0.norm3": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.norm1": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.norm2": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.norm3": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.norm1": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.norm2": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.norm3": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.norm1": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.LayerNorm",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.norm2": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.LayerNorm",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.norm3": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.LayerNorm",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_q": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_k": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn1.to_v": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_q": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_q": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_k": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn1.to_v": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_q": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_q": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_k": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn1.to_v": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_q": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_1",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_k": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.attn2.to_v": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_k": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"mid_block.attentions.0.transformer_blocks.0.attn2.to_v": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_k": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_2",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.attn2.to_v": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_2.Attention.Distribute.Linear_3",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.ff.net.0.proj": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"mid_block.attentions.0.transformer_blocks.0.ff.net.0.proj": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.ff.net.0.proj": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_1",
|
||||
"down_blocks.2.attentions.0.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_8.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"down_blocks.2.attentions.1.transformer_blocks.0.ff.net.2": "DownBlocks.Chain_9.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"mid_block.attentions.0.transformer_blocks.0.ff.net.2": "Sum.MiddleBlock.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.1.attentions.0.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_4.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.1.attentions.1.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_5.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.1.attentions.2.transformer_blocks.0.ff.net.2": "UpBlocks.Chain_6.CLIPLCrossAttention.Chain_2.CrossAttentionBlock.Residual_3.Linear_2",
|
||||
"up_blocks.0.resnets.0.norm1": "UpBlocks.Chain_1.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.0.resnets.1.norm1": "UpBlocks.Chain_2.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.0.resnets.2.norm1": "UpBlocks.Chain_3.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.1.resnets.0.norm1": "UpBlocks.Chain_4.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.1.resnets.1.norm1": "UpBlocks.Chain_5.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.0.resnets.0.conv1": "UpBlocks.Chain_1.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.0.resnets.1.conv1": "UpBlocks.Chain_2.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.0.resnets.2.conv1": "UpBlocks.Chain_3.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.1.resnets.0.conv1": "UpBlocks.Chain_4.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.1.resnets.1.conv1": "UpBlocks.Chain_5.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.0.resnets.0.conv_shortcut": "UpBlocks.Chain_1.ResidualBlock.Conv2d",
|
||||
"up_blocks.0.resnets.1.conv_shortcut": "UpBlocks.Chain_2.ResidualBlock.Conv2d",
|
||||
"up_blocks.0.resnets.2.conv_shortcut": "UpBlocks.Chain_3.ResidualBlock.Conv2d",
|
||||
"up_blocks.1.resnets.0.conv_shortcut": "UpBlocks.Chain_4.ResidualBlock.Conv2d",
|
||||
"up_blocks.1.resnets.1.conv_shortcut": "UpBlocks.Chain_5.ResidualBlock.Conv2d",
|
||||
"up_blocks.1.resnets.2.norm1": "UpBlocks.Chain_6.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.2.resnets.0.norm1": "UpBlocks.Chain_7.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.1.resnets.2.conv1": "UpBlocks.Chain_6.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.1.resnets.2.conv_shortcut": "UpBlocks.Chain_6.ResidualBlock.Conv2d",
|
||||
"up_blocks.2.resnets.0.conv1": "UpBlocks.Chain_7.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.2.resnets.0.conv_shortcut": "UpBlocks.Chain_7.ResidualBlock.Conv2d",
|
||||
"up_blocks.2.resnets.1.conv1": "UpBlocks.Chain_8.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.2.resnets.1.conv_shortcut": "UpBlocks.Chain_8.ResidualBlock.Conv2d",
|
||||
"up_blocks.2.resnets.2.norm1": "UpBlocks.Chain_9.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.3.resnets.0.norm1": "UpBlocks.Chain_10.ResidualBlock.Chain.GroupNorm_1",
|
||||
"up_blocks.2.resnets.2.conv1": "UpBlocks.Chain_9.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.2.resnets.2.conv_shortcut": "UpBlocks.Chain_9.ResidualBlock.Conv2d",
|
||||
"up_blocks.3.resnets.0.conv1": "UpBlocks.Chain_10.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.3.resnets.0.conv_shortcut": "UpBlocks.Chain_10.ResidualBlock.Conv2d",
|
||||
"up_blocks.3.resnets.1.conv1": "UpBlocks.Chain_11.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.3.resnets.2.conv1": "UpBlocks.Chain_12.ResidualBlock.Chain.RangeAdapter2d.Conv2d",
|
||||
"up_blocks.3.resnets.1.conv_shortcut": "UpBlocks.Chain_11.ResidualBlock.Conv2d",
|
||||
"up_blocks.3.resnets.2.conv_shortcut": "UpBlocks.Chain_12.ResidualBlock.Conv2d",
|
||||
"conv_out": "Chain.Conv2d"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,20 @@
|
||||
{
|
||||
"name_map": {
|
||||
"model0.1": "Chain_1.Conv2d",
|
||||
"model1.0": "Chain_2.Conv2d",
|
||||
"model1.3": "Chain_3.Conv2d",
|
||||
"model2.0.conv_block.1": "Residual_1.Conv2d_1",
|
||||
"model2.0.conv_block.5": "Residual_1.Conv2d_2",
|
||||
"model2.1.conv_block.1": "Residual_2.Conv2d_1",
|
||||
"model2.1.conv_block.5": "Residual_2.Conv2d_2",
|
||||
"model2.2.conv_block.1": "Residual_3.Conv2d_1",
|
||||
"model2.2.conv_block.5": "Residual_3.Conv2d_2",
|
||||
"model3.0": "Chain_4.ConvTranspose2d",
|
||||
"model3.3": "Chain_5.ConvTranspose2d",
|
||||
"model4.1": "Chain_6.Conv2d"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
@ -0,0 +1,275 @@
|
||||
{
|
||||
"name_map": {
|
||||
"vision_model.embeddings.patch_embedding": "ViTEmbeddings.Concatenate.Chain.PatchEncoder.Conv2d",
|
||||
"vision_model.embeddings.position_embedding": "ViTEmbeddings.Residual.PositionalEncoder.Embedding",
|
||||
"vision_model.embeddings.class_embedding": "ViTEmbeddings.Concatenate.ClassToken.Parameter.weight",
|
||||
"vision_model.pre_layrnorm": "LayerNorm_1",
|
||||
"vision_model.encoder.layers.0.layer_norm1": "Chain.TransformerLayer_1.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.0.layer_norm2": "Chain.TransformerLayer_1.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.1.layer_norm1": "Chain.TransformerLayer_2.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.1.layer_norm2": "Chain.TransformerLayer_2.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.2.layer_norm1": "Chain.TransformerLayer_3.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.2.layer_norm2": "Chain.TransformerLayer_3.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.3.layer_norm1": "Chain.TransformerLayer_4.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.3.layer_norm2": "Chain.TransformerLayer_4.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.4.layer_norm1": "Chain.TransformerLayer_5.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.4.layer_norm2": "Chain.TransformerLayer_5.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.5.layer_norm1": "Chain.TransformerLayer_6.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.5.layer_norm2": "Chain.TransformerLayer_6.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.6.layer_norm1": "Chain.TransformerLayer_7.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.6.layer_norm2": "Chain.TransformerLayer_7.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.7.layer_norm1": "Chain.TransformerLayer_8.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.7.layer_norm2": "Chain.TransformerLayer_8.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.8.layer_norm1": "Chain.TransformerLayer_9.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.8.layer_norm2": "Chain.TransformerLayer_9.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.9.layer_norm1": "Chain.TransformerLayer_10.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.9.layer_norm2": "Chain.TransformerLayer_10.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.10.layer_norm1": "Chain.TransformerLayer_11.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.10.layer_norm2": "Chain.TransformerLayer_11.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.11.layer_norm1": "Chain.TransformerLayer_12.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.11.layer_norm2": "Chain.TransformerLayer_12.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.12.layer_norm1": "Chain.TransformerLayer_13.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.12.layer_norm2": "Chain.TransformerLayer_13.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.13.layer_norm1": "Chain.TransformerLayer_14.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.13.layer_norm2": "Chain.TransformerLayer_14.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.14.layer_norm1": "Chain.TransformerLayer_15.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.14.layer_norm2": "Chain.TransformerLayer_15.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.15.layer_norm1": "Chain.TransformerLayer_16.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.15.layer_norm2": "Chain.TransformerLayer_16.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.16.layer_norm1": "Chain.TransformerLayer_17.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.16.layer_norm2": "Chain.TransformerLayer_17.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.17.layer_norm1": "Chain.TransformerLayer_18.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.17.layer_norm2": "Chain.TransformerLayer_18.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.18.layer_norm1": "Chain.TransformerLayer_19.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.18.layer_norm2": "Chain.TransformerLayer_19.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.19.layer_norm1": "Chain.TransformerLayer_20.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.19.layer_norm2": "Chain.TransformerLayer_20.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.20.layer_norm1": "Chain.TransformerLayer_21.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.20.layer_norm2": "Chain.TransformerLayer_21.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.21.layer_norm1": "Chain.TransformerLayer_22.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.21.layer_norm2": "Chain.TransformerLayer_22.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.22.layer_norm1": "Chain.TransformerLayer_23.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.22.layer_norm2": "Chain.TransformerLayer_23.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.23.layer_norm1": "Chain.TransformerLayer_24.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.23.layer_norm2": "Chain.TransformerLayer_24.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.24.layer_norm1": "Chain.TransformerLayer_25.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.24.layer_norm2": "Chain.TransformerLayer_25.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.25.layer_norm1": "Chain.TransformerLayer_26.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.25.layer_norm2": "Chain.TransformerLayer_26.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.26.layer_norm1": "Chain.TransformerLayer_27.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.26.layer_norm2": "Chain.TransformerLayer_27.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.27.layer_norm1": "Chain.TransformerLayer_28.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.27.layer_norm2": "Chain.TransformerLayer_28.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.28.layer_norm1": "Chain.TransformerLayer_29.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.28.layer_norm2": "Chain.TransformerLayer_29.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.29.layer_norm1": "Chain.TransformerLayer_30.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.29.layer_norm2": "Chain.TransformerLayer_30.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.30.layer_norm1": "Chain.TransformerLayer_31.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.30.layer_norm2": "Chain.TransformerLayer_31.Residual_2.LayerNorm",
|
||||
"vision_model.encoder.layers.31.layer_norm1": "Chain.TransformerLayer_32.Residual_1.LayerNorm",
|
||||
"vision_model.encoder.layers.31.layer_norm2": "Chain.TransformerLayer_32.Residual_2.LayerNorm",
|
||||
"vision_model.post_layernorm": "LayerNorm_2",
|
||||
"vision_model.encoder.layers.0.self_attn.q_proj": "Chain.TransformerLayer_1.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.0.self_attn.k_proj": "Chain.TransformerLayer_1.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.0.self_attn.v_proj": "Chain.TransformerLayer_1.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.0.self_attn.out_proj": "Chain.TransformerLayer_1.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.1.self_attn.q_proj": "Chain.TransformerLayer_2.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.1.self_attn.k_proj": "Chain.TransformerLayer_2.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.1.self_attn.v_proj": "Chain.TransformerLayer_2.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.1.self_attn.out_proj": "Chain.TransformerLayer_2.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.2.self_attn.q_proj": "Chain.TransformerLayer_3.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.2.self_attn.k_proj": "Chain.TransformerLayer_3.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.2.self_attn.v_proj": "Chain.TransformerLayer_3.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.2.self_attn.out_proj": "Chain.TransformerLayer_3.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.3.self_attn.q_proj": "Chain.TransformerLayer_4.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.3.self_attn.k_proj": "Chain.TransformerLayer_4.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.3.self_attn.v_proj": "Chain.TransformerLayer_4.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.3.self_attn.out_proj": "Chain.TransformerLayer_4.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.4.self_attn.q_proj": "Chain.TransformerLayer_5.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.4.self_attn.k_proj": "Chain.TransformerLayer_5.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.4.self_attn.v_proj": "Chain.TransformerLayer_5.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.4.self_attn.out_proj": "Chain.TransformerLayer_5.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.5.self_attn.q_proj": "Chain.TransformerLayer_6.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.5.self_attn.k_proj": "Chain.TransformerLayer_6.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.5.self_attn.v_proj": "Chain.TransformerLayer_6.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.5.self_attn.out_proj": "Chain.TransformerLayer_6.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.6.self_attn.q_proj": "Chain.TransformerLayer_7.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.6.self_attn.k_proj": "Chain.TransformerLayer_7.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.6.self_attn.v_proj": "Chain.TransformerLayer_7.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.6.self_attn.out_proj": "Chain.TransformerLayer_7.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.7.self_attn.q_proj": "Chain.TransformerLayer_8.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.7.self_attn.k_proj": "Chain.TransformerLayer_8.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.7.self_attn.v_proj": "Chain.TransformerLayer_8.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.7.self_attn.out_proj": "Chain.TransformerLayer_8.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.8.self_attn.q_proj": "Chain.TransformerLayer_9.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.8.self_attn.k_proj": "Chain.TransformerLayer_9.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.8.self_attn.v_proj": "Chain.TransformerLayer_9.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.8.self_attn.out_proj": "Chain.TransformerLayer_9.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.9.self_attn.q_proj": "Chain.TransformerLayer_10.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.9.self_attn.k_proj": "Chain.TransformerLayer_10.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.9.self_attn.v_proj": "Chain.TransformerLayer_10.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.9.self_attn.out_proj": "Chain.TransformerLayer_10.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.10.self_attn.q_proj": "Chain.TransformerLayer_11.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.10.self_attn.k_proj": "Chain.TransformerLayer_11.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.10.self_attn.v_proj": "Chain.TransformerLayer_11.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.10.self_attn.out_proj": "Chain.TransformerLayer_11.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.11.self_attn.q_proj": "Chain.TransformerLayer_12.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.11.self_attn.k_proj": "Chain.TransformerLayer_12.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.11.self_attn.v_proj": "Chain.TransformerLayer_12.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.11.self_attn.out_proj": "Chain.TransformerLayer_12.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.12.self_attn.q_proj": "Chain.TransformerLayer_13.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.12.self_attn.k_proj": "Chain.TransformerLayer_13.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.12.self_attn.v_proj": "Chain.TransformerLayer_13.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.12.self_attn.out_proj": "Chain.TransformerLayer_13.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.13.self_attn.q_proj": "Chain.TransformerLayer_14.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.13.self_attn.k_proj": "Chain.TransformerLayer_14.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.13.self_attn.v_proj": "Chain.TransformerLayer_14.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.13.self_attn.out_proj": "Chain.TransformerLayer_14.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.14.self_attn.q_proj": "Chain.TransformerLayer_15.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.14.self_attn.k_proj": "Chain.TransformerLayer_15.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.14.self_attn.v_proj": "Chain.TransformerLayer_15.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.14.self_attn.out_proj": "Chain.TransformerLayer_15.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.15.self_attn.q_proj": "Chain.TransformerLayer_16.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.15.self_attn.k_proj": "Chain.TransformerLayer_16.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.15.self_attn.v_proj": "Chain.TransformerLayer_16.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.15.self_attn.out_proj": "Chain.TransformerLayer_16.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.16.self_attn.q_proj": "Chain.TransformerLayer_17.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.16.self_attn.k_proj": "Chain.TransformerLayer_17.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.16.self_attn.v_proj": "Chain.TransformerLayer_17.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.16.self_attn.out_proj": "Chain.TransformerLayer_17.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.17.self_attn.q_proj": "Chain.TransformerLayer_18.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.17.self_attn.k_proj": "Chain.TransformerLayer_18.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.17.self_attn.v_proj": "Chain.TransformerLayer_18.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.17.self_attn.out_proj": "Chain.TransformerLayer_18.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.18.self_attn.q_proj": "Chain.TransformerLayer_19.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.18.self_attn.k_proj": "Chain.TransformerLayer_19.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.18.self_attn.v_proj": "Chain.TransformerLayer_19.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.18.self_attn.out_proj": "Chain.TransformerLayer_19.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.19.self_attn.q_proj": "Chain.TransformerLayer_20.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.19.self_attn.k_proj": "Chain.TransformerLayer_20.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.19.self_attn.v_proj": "Chain.TransformerLayer_20.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.19.self_attn.out_proj": "Chain.TransformerLayer_20.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.20.self_attn.q_proj": "Chain.TransformerLayer_21.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.20.self_attn.k_proj": "Chain.TransformerLayer_21.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.20.self_attn.v_proj": "Chain.TransformerLayer_21.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.20.self_attn.out_proj": "Chain.TransformerLayer_21.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.21.self_attn.q_proj": "Chain.TransformerLayer_22.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.21.self_attn.k_proj": "Chain.TransformerLayer_22.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.21.self_attn.v_proj": "Chain.TransformerLayer_22.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.21.self_attn.out_proj": "Chain.TransformerLayer_22.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.22.self_attn.q_proj": "Chain.TransformerLayer_23.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.22.self_attn.k_proj": "Chain.TransformerLayer_23.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.22.self_attn.v_proj": "Chain.TransformerLayer_23.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.22.self_attn.out_proj": "Chain.TransformerLayer_23.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.23.self_attn.q_proj": "Chain.TransformerLayer_24.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.23.self_attn.k_proj": "Chain.TransformerLayer_24.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.23.self_attn.v_proj": "Chain.TransformerLayer_24.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.23.self_attn.out_proj": "Chain.TransformerLayer_24.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.24.self_attn.q_proj": "Chain.TransformerLayer_25.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.24.self_attn.k_proj": "Chain.TransformerLayer_25.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.24.self_attn.v_proj": "Chain.TransformerLayer_25.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.24.self_attn.out_proj": "Chain.TransformerLayer_25.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.25.self_attn.q_proj": "Chain.TransformerLayer_26.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.25.self_attn.k_proj": "Chain.TransformerLayer_26.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.25.self_attn.v_proj": "Chain.TransformerLayer_26.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.25.self_attn.out_proj": "Chain.TransformerLayer_26.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.26.self_attn.q_proj": "Chain.TransformerLayer_27.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.26.self_attn.k_proj": "Chain.TransformerLayer_27.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.26.self_attn.v_proj": "Chain.TransformerLayer_27.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.26.self_attn.out_proj": "Chain.TransformerLayer_27.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.27.self_attn.q_proj": "Chain.TransformerLayer_28.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.27.self_attn.k_proj": "Chain.TransformerLayer_28.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.27.self_attn.v_proj": "Chain.TransformerLayer_28.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.27.self_attn.out_proj": "Chain.TransformerLayer_28.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.28.self_attn.q_proj": "Chain.TransformerLayer_29.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.28.self_attn.k_proj": "Chain.TransformerLayer_29.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.28.self_attn.v_proj": "Chain.TransformerLayer_29.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.28.self_attn.out_proj": "Chain.TransformerLayer_29.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.29.self_attn.q_proj": "Chain.TransformerLayer_30.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.29.self_attn.k_proj": "Chain.TransformerLayer_30.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.29.self_attn.v_proj": "Chain.TransformerLayer_30.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.29.self_attn.out_proj": "Chain.TransformerLayer_30.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.30.self_attn.q_proj": "Chain.TransformerLayer_31.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.30.self_attn.k_proj": "Chain.TransformerLayer_31.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.30.self_attn.v_proj": "Chain.TransformerLayer_31.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.30.self_attn.out_proj": "Chain.TransformerLayer_31.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.31.self_attn.q_proj": "Chain.TransformerLayer_32.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"vision_model.encoder.layers.31.self_attn.k_proj": "Chain.TransformerLayer_32.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"vision_model.encoder.layers.31.self_attn.v_proj": "Chain.TransformerLayer_32.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"vision_model.encoder.layers.31.self_attn.out_proj": "Chain.TransformerLayer_32.Residual_1.SelfAttention.Linear",
|
||||
"vision_model.encoder.layers.0.mlp.fc1": "Chain.TransformerLayer_1.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.1.mlp.fc1": "Chain.TransformerLayer_2.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.2.mlp.fc1": "Chain.TransformerLayer_3.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.3.mlp.fc1": "Chain.TransformerLayer_4.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.4.mlp.fc1": "Chain.TransformerLayer_5.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.5.mlp.fc1": "Chain.TransformerLayer_6.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.6.mlp.fc1": "Chain.TransformerLayer_7.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.7.mlp.fc1": "Chain.TransformerLayer_8.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.8.mlp.fc1": "Chain.TransformerLayer_9.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.9.mlp.fc1": "Chain.TransformerLayer_10.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.10.mlp.fc1": "Chain.TransformerLayer_11.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.11.mlp.fc1": "Chain.TransformerLayer_12.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.12.mlp.fc1": "Chain.TransformerLayer_13.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.13.mlp.fc1": "Chain.TransformerLayer_14.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.14.mlp.fc1": "Chain.TransformerLayer_15.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.15.mlp.fc1": "Chain.TransformerLayer_16.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.16.mlp.fc1": "Chain.TransformerLayer_17.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.17.mlp.fc1": "Chain.TransformerLayer_18.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.18.mlp.fc1": "Chain.TransformerLayer_19.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.19.mlp.fc1": "Chain.TransformerLayer_20.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.20.mlp.fc1": "Chain.TransformerLayer_21.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.21.mlp.fc1": "Chain.TransformerLayer_22.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.22.mlp.fc1": "Chain.TransformerLayer_23.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.23.mlp.fc1": "Chain.TransformerLayer_24.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.24.mlp.fc1": "Chain.TransformerLayer_25.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.25.mlp.fc1": "Chain.TransformerLayer_26.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.26.mlp.fc1": "Chain.TransformerLayer_27.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.27.mlp.fc1": "Chain.TransformerLayer_28.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.28.mlp.fc1": "Chain.TransformerLayer_29.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.29.mlp.fc1": "Chain.TransformerLayer_30.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.30.mlp.fc1": "Chain.TransformerLayer_31.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.31.mlp.fc1": "Chain.TransformerLayer_32.Residual_2.FeedForward.Linear_1",
|
||||
"vision_model.encoder.layers.0.mlp.fc2": "Chain.TransformerLayer_1.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.1.mlp.fc2": "Chain.TransformerLayer_2.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.2.mlp.fc2": "Chain.TransformerLayer_3.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.3.mlp.fc2": "Chain.TransformerLayer_4.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.4.mlp.fc2": "Chain.TransformerLayer_5.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.5.mlp.fc2": "Chain.TransformerLayer_6.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.6.mlp.fc2": "Chain.TransformerLayer_7.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.7.mlp.fc2": "Chain.TransformerLayer_8.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.8.mlp.fc2": "Chain.TransformerLayer_9.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.9.mlp.fc2": "Chain.TransformerLayer_10.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.10.mlp.fc2": "Chain.TransformerLayer_11.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.11.mlp.fc2": "Chain.TransformerLayer_12.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.12.mlp.fc2": "Chain.TransformerLayer_13.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.13.mlp.fc2": "Chain.TransformerLayer_14.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.14.mlp.fc2": "Chain.TransformerLayer_15.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.15.mlp.fc2": "Chain.TransformerLayer_16.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.16.mlp.fc2": "Chain.TransformerLayer_17.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.17.mlp.fc2": "Chain.TransformerLayer_18.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.18.mlp.fc2": "Chain.TransformerLayer_19.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.19.mlp.fc2": "Chain.TransformerLayer_20.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.20.mlp.fc2": "Chain.TransformerLayer_21.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.21.mlp.fc2": "Chain.TransformerLayer_22.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.22.mlp.fc2": "Chain.TransformerLayer_23.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.23.mlp.fc2": "Chain.TransformerLayer_24.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.24.mlp.fc2": "Chain.TransformerLayer_25.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.25.mlp.fc2": "Chain.TransformerLayer_26.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.26.mlp.fc2": "Chain.TransformerLayer_27.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.27.mlp.fc2": "Chain.TransformerLayer_28.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.28.mlp.fc2": "Chain.TransformerLayer_29.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.29.mlp.fc2": "Chain.TransformerLayer_30.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.30.mlp.fc2": "Chain.TransformerLayer_31.Residual_2.FeedForward.Linear_2",
|
||||
"vision_model.encoder.layers.31.mlp.fc2": "Chain.TransformerLayer_32.Residual_2.FeedForward.Linear_2",
|
||||
"visual_projection": "Linear"
|
||||
},
|
||||
"regex_map": {},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {
|
||||
"ViTEmbeddings.Concatenate.ClassToken.Parameter.weight": [
|
||||
1,
|
||||
1280
|
||||
]
|
||||
}
|
||||
}
|
@ -0,0 +1,19 @@
|
||||
{
|
||||
"name_map": {
|
||||
"text_model.embeddings.token_embedding": "Sum.TokenEncoder",
|
||||
"text_model.embeddings.position_embedding": "Sum.PositionalEncoder.Embedding",
|
||||
"text_model.final_layer_norm": "LayerNorm",
|
||||
"text_projection": "Linear"
|
||||
},
|
||||
"regex_map": {
|
||||
"text_model\\.encoder\\.layers\\.(?P<layer>\\d+)\\.layer_norm(?P<norm>\\d+)": "TransformerLayer_{int(layer) + 1}.Residual_{norm}.LayerNorm",
|
||||
"text_model\\.encoder\\.layers\\.(?P<layer_number>\\d+)\\.self_attn\\.q_proj": "TransformerLayer_{int(layer_number) + 1}.Residual_1.SelfAttention.Distribute.Linear_1",
|
||||
"text_model\\.encoder\\.layers\\.(?P<layer_number>\\d+)\\.self_attn\\.k_proj": "TransformerLayer_{int(layer_number) + 1}.Residual_1.SelfAttention.Distribute.Linear_2",
|
||||
"text_model\\.encoder\\.layers\\.(?P<layer_number>\\d+)\\.self_attn\\.v_proj": "TransformerLayer_{int(layer_number) + 1}.Residual_1.SelfAttention.Distribute.Linear_3",
|
||||
"text_model\\.encoder\\.layers\\.(?P<layer_number>\\d+)\\.self_attn\\.out_proj": "TransformerLayer_{int(layer_number) + 1}.Residual_1.SelfAttention.Linear",
|
||||
"text_model\\.encoder\\.layers\\.(?P<layer_number>\\d+)\\.mlp\\.fc(?P<fc_num>\\d+)": "TransformerLayer_{int(layer_number) + 1}.Residual_2.FeedForward.Linear_{fc_num}"
|
||||
},
|
||||
"ignore_prefixes": [],
|
||||
"source_aliases": {},
|
||||
"reshapes": {}
|
||||
}
|
|
@ -0,0 +1,2 @@
|
||||
def test_sdxl_vae():
|
||||
pass
|
Loading…
Reference in New Issue