From 64ec53b5a6e4b73a9618ac1ea66b9f2fb9f7d54a Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Thu, 18 Jan 2024 09:52:29 +0100 Subject: [PATCH] add comment --- src/refiners/foundationals/latent_diffusion/lora.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/refiners/foundationals/latent_diffusion/lora.py b/src/refiners/foundationals/latent_diffusion/lora.py index 92af022c9..d25115d20 100644 --- a/src/refiners/foundationals/latent_diffusion/lora.py +++ b/src/refiners/foundationals/latent_diffusion/lora.py @@ -109,6 +109,7 @@ def pad(input: str, /, padding_length: int = 2) -> str: @staticmethod def sort_keys(key: str, /) -> tuple[str, int]: + # out0 happens sometimes as an alias for out ; this dict might not be exhaustive key_char_order = {"q": 1, "k": 2, "v": 3, "out": 4, "out0": 4} for i, s in enumerate(key.split("_")):