removed redundant hqq config validation
This commit is contained in:
committed by
Sung Ching Liu
parent
ba8e29c841
commit
098ffcc5a2
@@ -377,18 +377,6 @@ class AxolotlInputConfig(
|
|||||||
raise ValueError(f"Only one of {', '.join(fields)} must be set")
|
raise ValueError(f"Only one of {', '.join(fields)} must be set")
|
||||||
return data
|
return data
|
||||||
|
|
||||||
@model_validator(mode="before")
|
|
||||||
@classmethod
|
|
||||||
def check_hqq_config_redundancy(cls, data):
|
|
||||||
if (data.get("load_in_4bit") or data.get("load_in_8bit")) and data.get(
|
|
||||||
"use_hqq"
|
|
||||||
):
|
|
||||||
|
|
||||||
raise ValueError(
|
|
||||||
"Can't simultaneously set `hqq` configurations and `load_in_4bit` / `load_in_8bit`"
|
|
||||||
)
|
|
||||||
return data
|
|
||||||
|
|
||||||
@model_validator(mode="before")
|
@model_validator(mode="before")
|
||||||
@classmethod
|
@classmethod
|
||||||
def check_batch_size_fields(cls, data):
|
def check_batch_size_fields(cls, data):
|
||||||
|
|||||||
Reference in New Issue
Block a user