support layerwise galore

Former-commit-id: 8664262cde
This commit is contained in:
hiyouga
2024-03-10 00:24:11 +08:00
parent 276def1897
commit 4a4e4b4354
14 changed files with 109 additions and 51 deletions

View File

@@ -182,6 +182,10 @@ class GaloreArguments:
default="std",
metadata={"help": "Type of GaLore projection."},
)
galore_layerwise: bool = field(
default=False,
metadata={"help": "Whether or not to enable layer-wise update to further save memory."},
)
@dataclass