298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

This is useless for inference.

Browse Source
This commit is contained in:
comfyanonymous 2023-05-31 13:03:24 -04:00
parent 606446d030
commit 03da8a3426
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/sd.py
View File

@ -743,7 +743,7 @@ def load_controlnet(ckpt_path, model=None):
use_spatial_transformer=True, use_spatial_transformer=True,
transformer_depth=1, transformer_depth=1,
context_dim=context_dim, context_dim=context_dim,
use_checkpoint=True, use_checkpoint=False,
legacy=False, legacy=False,
use_fp16=use_fp16) use_fp16=use_fp16)
else: else:
@ -760,7 +760,7 @@ def load_controlnet(ckpt_path, model=None):
use_linear_in_transformer=True, use_linear_in_transformer=True,
transformer_depth=1, transformer_depth=1,
context_dim=context_dim, context_dim=context_dim,
use_checkpoint=True, use_checkpoint=False,
legacy=False, legacy=False,
use_fp16=use_fp16) use_fp16=use_fp16)
if pth: if pth:
@ -1045,7 +1045,7 @@ def load_checkpoint_guess_config(ckpt_path, output_vae=True, output_clip=True, o
} }
unet_config = { unet_config = {
"use_checkpoint": True, "use_checkpoint": False,
"image_size": 32, "image_size": 32,
"out_channels": 4, "out_channels": 4,
"attention_resolutions": [ "attention_resolutions": [