fix

mosaicml · Aug 15, 2024 · c9d1884 · c9d1884
1 parent 11eab11
commit c9d1884
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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}