From 8be8c85a631ed347409ffb1522b82e1b0a613c35 Mon Sep 17 00:00:00 2001 From: Srini Iyer Date: Fri, 14 Feb 2025 19:20:17 +0000 Subject: [PATCH] black --- bytelatent/base_transformer.py | 2 +- bytelatent/model/latent_transformer.py | 2 +- bytelatent/model/local_models.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/bytelatent/base_transformer.py b/bytelatent/base_transformer.py index eed4912..7b76b9e 100644 --- a/bytelatent/base_transformer.py +++ b/bytelatent/base_transformer.py @@ -22,7 +22,7 @@ try: RMSNorm = FusedRMSNorm except (ImportError, ModuleNotFoundError): - print('Apex not found. Using nn.RMSNorm') + print("Apex not found. Using nn.RMSNorm") RMSNorm = nn.RMSNorm if int(os.environ.get("BLT_ALLOW_MISSING_FLEX_ATTENTION", False)) == 0: diff --git a/bytelatent/model/latent_transformer.py b/bytelatent/model/latent_transformer.py index 1e615b6..95b6d8b 100644 --- a/bytelatent/model/latent_transformer.py +++ b/bytelatent/model/latent_transformer.py @@ -22,7 +22,7 @@ try: RMSNorm = FusedRMSNorm except (ImportError, ModuleNotFoundError): - print('Apex not found. Using nn.RMSNorm') + print("Apex not found. Using nn.RMSNorm") RMSNorm = nn.RMSNorm logger = logging.getLogger() diff --git a/bytelatent/model/local_models.py b/bytelatent/model/local_models.py index 1aba0e8..353c878 100644 --- a/bytelatent/model/local_models.py +++ b/bytelatent/model/local_models.py @@ -26,7 +26,7 @@ try: RMSNorm = FusedRMSNorm except (ImportError, ModuleNotFoundError): - print('Apex not found. Using nn.RMSNorm') + print("Apex not found. Using nn.RMSNorm") RMSNorm = nn.RMSNorm logger = logging.getLogger()