Update src/peft/tuners/adalora.py

Co-authored-by: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
This commit is contained in:
Qingru Zhang
2023-04-05 16:30:16 -04:00
committed by GitHub
co-authored by Sourab Mangrulkar
parent 3e6a88a8f9
commit b3e6ef6224
+1 -1
View File
@@ -127,7 +127,7 @@ class AdaLoraModel(LoraModel):
is_target_modules_in_base_model = True
parent, target, target_name = self._get_submodules(key)
bias = target.bias is not None
if loaded_in_8bit and isinstance(target, bnb.nn.Linear8bitLt) and self.peft_config.enable_lora is None:
if loaded_in_8bit and isinstance(target, bnb.nn.Linear8bitLt):
kwargs.update(
{
"has_fp16_weights": target.state.has_fp16_weights,