fix eetq peft

This commit is contained in:
Younes Belkada
2024-04-29 18:12:40 +02:00
parent e19f7bf424
commit d34b1b43ce

View File

@ -96,6 +96,7 @@ def dispatch_eetq(
if is_eetq_available() and isinstance(target_base_layer, EetqLinear):
new_module = EetqLoraLinear(target, adapter_name, **kwargs)
new_module.train()
target.weight = target_base_layer.weight
if hasattr(target, "bias"):