From 4376b125ebc7736d0db67f0326027a4dbf7c9659 Mon Sep 17 00:00:00 2001 From: comfyanonymous Date: Thu, 29 Jun 2023 00:26:33 -0400 Subject: [PATCH] Remove useless code. --- comfy/model_detection.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/comfy/model_detection.py b/comfy/model_detection.py index 48137c78..edad48b1 100644 --- a/comfy/model_detection.py +++ b/comfy/model_detection.py @@ -16,13 +16,11 @@ def count_blocks(state_dict_keys, prefix_string): def detect_unet_config(state_dict, key_prefix, use_fp16): state_dict_keys = list(state_dict.keys()) - num_res_blocks = 2 unet_config = { "use_checkpoint": False, "image_size": 32, "out_channels": 4, - "num_res_blocks": num_res_blocks, "use_spatial_transformer": True, "legacy": False }