Add ability to train a full rank LoRA. (experimental)
This commit is contained in:
1 parent
645046701b
commit
af6fdaaaf9
3 files changed
+34
-8
No files matched your search
@@ -157,6 +157,9 @@ class NetworkConfig:
|
||||
elif linear is not None:
|
||||
self.rank: int = linear
|
||||
self.linear: int = linear
|
||||
else:
|
||||
self.rank: int = 4
|
||||
self.linear: int = 4
|
||||
self.conv: int = kwargs.get('conv', None)
|
||||
self.alpha: float = kwargs.get('alpha', 1.0)
|
||||
self.linear_alpha: float = kwargs.get('linear_alpha', self.alpha)
|
||||
|
||||
Reference in new issue
Block a user