from local

This commit is contained in:
dohe0342 2023-05-18 16:57:07 +09:00
parent a49f8cfa48
commit fbb9649217
2 changed files with 1 additions and 1 deletions

View File

@ -238,7 +238,7 @@ class LoRAModule(nn.Module):
self, self,
embedding_dim: float = 768, embedding_dim: float = 768,
layer_num: int = 12, layer_num: int = 12,
rank: int = 10, rank: int = 16,
) -> None: ) -> None:
super().__init__() super().__init__()