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

simplify is_compatible in lora.py

Browse source
This commit is contained in:
Laurent 2024-02-14 15:20:41 +00:00 committed by Laureηt
parent 35868ba34b
commit 0230971543
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/refiners/fluxion/adapters/lora.py
View file

@ -250,9 +250,11 @@ class LinearLora(Lora[fl.Linear]):
) )
def is_compatible(self, layer: fl.WeightedModule, /) -> bool: def is_compatible(self, layer: fl.WeightedModule, /) -> bool:
if isinstance(layer, fl.Linear): return (
return layer.in_features == self.in_features and layer.out_features == self.out_features isinstance(layer, fl.Linear)
return False and layer.in_features == self.in_features
and layer.out_features == self.out_features
)
class Conv2dLora(Lora[fl.Conv2d]): class Conv2dLora(Lora[fl.Conv2d]):