Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-09 15:02:01 +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:
if isinstance(layer, fl.Linear):
return layer.in_features == self.in_features and layer.out_features == self.out_features
return False
return (
isinstance(layer, fl.Linear)
and layer.in_features == self.in_features
and layer.out_features == self.out_features
)
class Conv2dLora(Lora[fl.Conv2d]):