mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-09 04:52:32 +08:00
delete debug print
This commit is contained in:
parent
9e006c3fa3
commit
3711e9462f
@ -258,7 +258,6 @@ def detect_unet_config(state_dict, key_prefix, metadata=None):
|
||||
dit_config["nerf_final_head_type"] = "conv" if f"{key_prefix}nerf_final_layer_conv.norm.scale" in state_dict_keys else "linear"
|
||||
dit_config["nerf_embedder_dtype"] = torch.float32
|
||||
if f"__x0__" in state_dict_keys: # x0 pred
|
||||
print("radiance in x0 mode")
|
||||
dit_config["use_x0"] = True
|
||||
else:
|
||||
dit_config["guidance_embed"] = "{}guidance_in.in_layer.weight".format(key_prefix) in state_dict_keys
|
||||
|
||||
Loading…
Reference in New Issue
Block a user