diff --git a/generative/networks/nets/autoencoderkl.py b/generative/networks/nets/autoencoderkl.py
index 7989d13f..23750839 100644
--- a/generative/networks/nets/autoencoderkl.py
+++ b/generative/networks/nets/autoencoderkl.py
@@ -613,7 +613,8 @@ def __init__(
 
         if len(num_res_blocks) != len(num_channels):
             raise ValueError(
-                "`num_res_blocks` should be a single integer or a tuple of integers with the same length as `num_channels`."
+                "`num_res_blocks` should be a single integer or a tuple of integers with the same length as "
+                "`num_channels`."
             )
 
         self.encoder = Encoder(
diff --git a/generative/networks/nets/diffusion_model_unet.py b/generative/networks/nets/diffusion_model_unet.py
index 54dfa277..b651f206 100644
--- a/generative/networks/nets/diffusion_model_unet.py
+++ b/generative/networks/nets/diffusion_model_unet.py
@@ -1656,7 +1656,8 @@ def __init__(
 
         if len(num_res_blocks) != len(num_channels):
             raise ValueError(
-                "`num_res_blocks` should be a single integer or a tuple of integers with the same length as `num_channels`."
+                "`num_res_blocks` should be a single integer or a tuple of integers with the same length as "
+                "`num_channels`."
             )
 
         self.in_channels = in_channels