Fix format (#251)

Signed-off-by: Walter Hugo Lopez Pinaya <ianonimato@hotmail.com>

Author: Warvito

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(

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