diff --git a/llmfoundry/callbacks/env_logging_callback.py b/llmfoundry/callbacks/env_logging_callback.py index 1c55fb15e8..1b5514823c 100644 --- a/llmfoundry/callbacks/env_logging_callback.py +++ b/llmfoundry/callbacks/env_logging_callback.py @@ -122,8 +122,9 @@ def _get_nvidia_info(self) -> dict[str, Any]: return { 'cuda_version': torch.version.cuda, # type: ignore[attr-defined] - 'cudnn_version': str(torch.backends.cudnn.version() - ), # type: ignore[attr-defined] + 'cudnn_version': str( + torch.backends.cudnn.version(), + ), # type: ignore[attr-defined] 'nccl_version': '.'.join(map(str, nccl_version)), } return {'available': False}