tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix attribute error

Browse Source
This commit is contained in:
Sunny Liu
2025-04-21 22:29:24 -04:00
parent c4910da015
commit 0179021780
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/axolotl/utils/schemas/peft.py
View File

@@ -87,24 +87,24 @@ class LoraConfig(BaseModel):
def validate_qlora(self): def validate_qlora(self):
if self.adapter == "qlora": if self.adapter == "qlora":
if self.merge_lora: if self.merge_lora:
if self.quantization.bits == 8 or self.cfg.load_in_8bit: if self.quantization.bits == 8 or self.load_in_8bit:
raise ValueError("Can't merge qlora if loaded in 8bit") raise ValueError("Can't merge qlora if loaded in 8bit")
if self.quantization.backend == "gptq": if self.quantization.backend == "gptq":
raise ValueError("Can't merge qlora if using gptq") raise ValueError("Can't merge qlora if using gptq")
if self.quantization.bits == 4 or self.cfg.load_in_4bit: if self.quantization.bits == 4 or self.load_in_4bit:
raise ValueError("Can't merge qlora if loaded in 4bit") raise ValueError("Can't merge qlora if loaded in 4bit")
else: else:
if self.quantization: if self.quantization:
if self.quantization.bits == 8 or self.cfg.load_in_8bit: if self.quantization.bits == 8 or self.load_in_8bit:
raise ValueError("Can't load qlora in 8bit") raise ValueError("Can't load qlora in 8bit")
if self.quantization.backend == "gptq": if self.quantization.backend == "gptq":
raise ValueError("Can't load qlora if using gptq") raise ValueError("Can't load qlora if using gptq")
if not self.quantization.bits == 4 or self.cfg.load_in_4bit: if not self.quantization.bits == 4 or self.load_in_4bit:
raise ValueError("Require quantization.bits <= 4 for qlora") raise ValueError("Require quantization.bits <= 4 for qlora")
return self return self