Skip to content

GitLab

Commit 03da8a34 authored by comfyanonymous's avatar comfyanonymous
Browse files

This is useless for inference.

parent 606446d0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
comfy/sd.py
View file @ 03da8a34
...@@ -743,7 +743,7 @@ def load_controlnet(ckpt_path, model=None): ...@@ -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): ...@@ -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 ...@@ -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": [
......
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