Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-14 00:58:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update src/refiners/foundationals/latent_diffusion/lora.py

Browse source 
Co-authored-by: Laureηt <laurent@lagon.tech>
This commit is contained in:
Pierre Chapuis 2024-02-07 14:35:22 +01:00
parent eb9abefe07
commit b1c200c63a
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/foundationals/latent_diffusion/lora.py
View file

@ -245,8 +245,8 @@ class SDLoraManager:
# get the suffix and score for `key` (default: no suffix, highest score = 5) # get the suffix and score for `key` (default: no suffix, highest score = 5)
(sfx, score) = next(((k, v) for k, v in key_char_order.items() if key.endswith(k)), ("", 5)) (sfx, score) = next(((k, v) for k, v in key_char_order.items() if key.endswith(k)), ("", 5))
# return tuple of: (padded key prefix, score) padded_key_prefix = SDLoraManager.pad(key.removesuffix(sfx))
return (SDLoraManager.pad(key.removesuffix(sfx)), score) return (padded_key_prefix, score)
@staticmethod @staticmethod
def auto_attach( def auto_attach(