qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove useless code.

This commit is contained in:
comfyanonymous 2023-06-29 00:26:33 -04:00
parent 89120f1fbe
commit 4376b125eb
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_detection.py
View File

@ -16,13 +16,11 @@ def count_blocks(state_dict_keys, prefix_string):
def detect_unet_config(state_dict, key_prefix, use_fp16): def detect_unet_config(state_dict, key_prefix, use_fp16):
state_dict_keys = list(state_dict.keys()) state_dict_keys = list(state_dict.keys())
num_res_blocks = 2
unet_config = { unet_config = {
"use_checkpoint": False, "use_checkpoint": False,
"image_size": 32, "image_size": 32,
"out_channels": 4, "out_channels": 4,
"num_res_blocks": num_res_blocks,
"use_spatial_transformer": True, "use_spatial_transformer": True,
"legacy": False "legacy": False
} }