fix trainable params issue

This commit is contained in:
Sourab Mangrulkar
2023-04-08 11:32:00 +05:30
parent e8b0085d2b
commit 7b7038273a
2 changed files with 2 additions and 3 deletions
+1 -1
View File
@@ -117,11 +117,11 @@ class AdaLoraModel(LoraModel):
"When using multiple adapters, set inference_mode to True for all adapters except the one you want to train."
)
mark_only_lora_as_trainable(self.model, self.peft_config[adapter_name].bias)
if self.peft_config[adapter_name].inference_mode:
_freeze_adapter(self.model, adapter_name)
else:
self.trainable_adapter_name = adapter_name
mark_only_lora_as_trainable(self.model, self.peft_config[adapter_name].bias)
self.rankallocator = RankAllocator(self.model, self.peft_config[adapter_name], self.trainable_adapter_name)
def _find_and_replace(self, adapter_name):
+1 -2
View File
@@ -143,10 +143,9 @@ class LoraModel(torch.nn.Module):
raise ValueError(
"LoraModel supports only 1 adapter with bias. When using multiple adapters, set bias to 'none' for all adapters."
)
mark_only_lora_as_trainable(self.model, self.peft_config[adapter_name].bias)
if self.peft_config[adapter_name].inference_mode:
_freeze_adapter(self.model, adapter_name)
else:
mark_only_lora_as_trainable(self.model, self.peft_config[adapter_name].bias)
def _find_and_replace(self, adapter_name):
lora_config = self.peft_config[adapter_name]