diff --git a/src/axolotl/utils/schemas/config.py b/src/axolotl/utils/schemas/config.py index 2a85d6c73..5c3357d72 100644 --- a/src/axolotl/utils/schemas/config.py +++ b/src/axolotl/utils/schemas/config.py @@ -1352,8 +1352,8 @@ class AxolotlConfigWCapabilities(AxolotlInputConfig): and not self.is_preprocess and (self.bf16 is True or self.bfloat16 is True) ): - raise ValueError( - "bf16 requested, but AMP is not supported on this GPU. Requires Ampere series or above." + LOG.warning( + "bf16 requested, but AMP is not supported on this GPU. Requires Ampere series or above. Training will fail, but other operations (such as merging) are still functional." ) return self diff --git a/tests/patched/test_validation.py b/tests/patched/test_validation.py index 29ab859c1..71400c4ae 100644 --- a/tests/patched/test_validation.py +++ b/tests/patched/test_validation.py @@ -726,8 +726,12 @@ class TestValidation(BaseValidation): | minimal_cfg ) - with pytest.raises(ValueError, match=r".*AMP is not supported on this GPU*"): + with self._caplog.at_level("WARNING"): AxolotlConfigWCapabilities(**cfg.to_dict()) + assert any( + "AMP is not supported" in record.message + for record in self._caplog.records + ) cfg = ( DictDefault(