diff --git a/comfy/utils.py b/comfy/utils.py
index c4c15def5..956ac1773 100644
--- a/comfy/utils.py
+++ b/comfy/utils.py
@@ -140,8 +140,10 @@ def unet_to_diffusers(unet_config):
     channel_mult = unet_config["channel_mult"]
     transformer_depth = unet_config["transformer_depth"]
     num_blocks = len(channel_mult)
-    if not isinstance(num_res_blocks, list):
+    if isinstance(num_res_blocks, int):
         num_res_blocks = [num_res_blocks] * num_blocks
+    if isinstance(transformer_depth, int):
+        transformer_depth = [transformer_depth] * num_blocks
 
     transformers_per_layer = []
     res = 1
@@ -152,7 +154,7 @@ def unet_to_diffusers(unet_config):
         transformers_per_layer.append(transformers)
         res *= 2
 
-    transformers_mid = unet_config.get("transformer_depth_middle", transformers_per_layer[-1])
+    transformers_mid = unet_config.get("transformer_depth_middle", transformer_depth[-1])
 
     diffusers_unet_map = {}
     for x in range(num_blocks):