optimize sd/paint_by_example modle VRAM usage

This commit is contained in:
Qing
2023-01-18 18:34:10 +08:00
parent 384f16dcd0
commit 148e97e8da
7 changed files with 61 additions and 24 deletions

View File

@@ -392,12 +392,12 @@ def main(args):
device=device,
no_half=args.no_half,
hf_access_token=args.hf_access_token,
sd_disable_nsfw=args.sd_disable_nsfw,
disable_nsfw=args.sd_disable_nsfw or args.disable_nsfw,
sd_cpu_textencoder=args.sd_cpu_textencoder,
sd_run_local=args.sd_run_local,
local_files_only=args.local_files_only,
cpu_offload=args.cpu_offload,
sd_enable_xformers=args.sd_enable_xformers,
enable_xformers=args.sd_enable_xformers or args.enable_xformers,
callback=diffuser_callback,
)