Skip to content

GitLab

Commit 4376b125 authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove useless code.

parent 89120f1f
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
comfy/model_detection.py
View file @ 4376b125
...@@ -16,13 +16,11 @@ def count_blocks(state_dict_keys, prefix_string): ...@@ -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
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment