Fix Latent Diffusion Inferer tests (#128)

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

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

Author: Warvito

tests/test_latent_diffusion_inferer.py

@@ -38,12 +38,11 @@
             "spatial_dims": 2,
             "in_channels": 3,
             "out_channels": 3,
-            "model_channels": 8,
+            "num_channels": [8, 8, 8],
             "norm_num_groups": 8,
-            "attention_resolutions": [8],
+            "attention_levels": [False, False, True],
             "num_res_blocks": 1,
-            "channel_mult": [1, 1, 1],
-            "num_heads": 1,
+            "num_head_channels": 8,
         },
         (1, 1, 32, 32),
         (1, 3, 8, 8),
@@ -67,12 +66,11 @@
             "spatial_dims": 2,
             "in_channels": 3,
             "out_channels": 3,
-            "model_channels": 8,
+            "num_channels": [8, 8, 8],
             "norm_num_groups": 8,
-            "attention_resolutions": [8],
+            "attention_levels": [False, False, True],
             "num_res_blocks": 1,
-            "channel_mult": [1, 1, 1],
-            "num_heads": 1,
+            "num_head_channels": 8,
         },
         (1, 1, 32, 32),
         (1, 3, 8, 8),