mirror of
https://github.com/lllyasviel/stable-diffusion-webui-forge.git
synced 2026-01-26 19:09:45 +00:00
b731bb860c
i Update initialization.py initialization initialization Update initialization.py i i Update sd_samplers_common.py Update sd_hijack.py i Update sd_models.py Update sd_models.py Update forge_loader.py Update sd_models.py i Update sd_model.py i Update sd_models.py Create sd_model.py i i Update sd_models.py i Update sd_models.py Update sd_models.py i i Update sd_samplers_common.py i Update sd_models.py Update sd_models.py Update sd_samplers_common.py Update sd_models.py Update sd_models.py Update sd_models.py Update sd_models.py Update sd_samplers_common.py i Update shared_options.py Update prompt_parser.py Update sd_hijack_unet.py i Update sd_models.py Update sd_models.py Update sd_models.py Update devices.py i Update sd_vae.py Update sd_models.py Update processing.py Update ui_settings.py Update sd_models_xl.py i i Update sd_samplers_kdiffusion.py Update sd_samplers_timesteps.py Update ui_settings.py Update cmd_args.py Update cmd_args.py Update initialization.py Update shared_options.py Update initialization.py Update shared_options.py i Update cmd_args.py Update initialization.py Update initialization.py Update initialization.py Update cmd_args.py Update cmd_args.py Update sd_hijack.py
67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
import argparse
|
|
|
|
|
|
def initialize_forge():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--no-vram", action="store_true")
|
|
parser.add_argument("--normal-vram", action="store_true")
|
|
parser.add_argument("--high-vram", action="store_true")
|
|
parser.add_argument("--always-vram", action="store_true")
|
|
vram_args = vars(parser.parse_known_args()[0])
|
|
|
|
parser = argparse.ArgumentParser()
|
|
attn_group = parser.add_mutually_exclusive_group()
|
|
attn_group.add_argument("--attention-split", action="store_true")
|
|
attn_group.add_argument("--attention-quad", action="store_true")
|
|
attn_group.add_argument("--attention-pytorch", action="store_true")
|
|
parser.add_argument("--disable-xformers", action="store_true")
|
|
fpte_group = parser.add_mutually_exclusive_group()
|
|
fpte_group.add_argument("--clip-in-fp8-e4m3fn", action="store_true")
|
|
fpte_group.add_argument("--clip-in-fp8-e5m2", action="store_true")
|
|
fpte_group.add_argument("--clip-in-fp16", action="store_true")
|
|
fpte_group.add_argument("--clip-in-fp32", action="store_true")
|
|
fp_group = parser.add_mutually_exclusive_group()
|
|
fp_group.add_argument("--all-in-fp32", action="store_true")
|
|
fp_group.add_argument("--all-in-fp16", action="store_true")
|
|
fpunet_group = parser.add_mutually_exclusive_group()
|
|
fpunet_group.add_argument("--unet-in-bf16", action="store_true")
|
|
fpunet_group.add_argument("--unet-in-fp16", action="store_true")
|
|
fpunet_group.add_argument("--unet-in-fp8-e4m3fn", action="store_true")
|
|
fpunet_group.add_argument("--unet-in-fp8-e5m2", action="store_true")
|
|
fpvae_group = parser.add_mutually_exclusive_group()
|
|
fpvae_group.add_argument("--vae-in-fp16", action="store_true")
|
|
fpvae_group.add_argument("--vae-in-fp32", action="store_true")
|
|
fpvae_group.add_argument("--vae-in-bf16", action="store_true")
|
|
other_args = vars(parser.parse_known_args()[0])
|
|
|
|
from ldm_patched.modules.args_parser import args
|
|
|
|
args.always_cpu = False
|
|
args.always_gpu = False
|
|
args.always_high_vram = False
|
|
args.always_low_vram = False
|
|
args.always_no_vram = False
|
|
args.always_offload_from_vram = True
|
|
args.async_cuda_allocation = False
|
|
args.disable_async_cuda_allocation = True
|
|
|
|
if vram_args['no_vram']:
|
|
args.always_cpu = True
|
|
|
|
if vram_args['always_vram']:
|
|
args.always_gpu = True
|
|
|
|
if vram_args['high_vram']:
|
|
args.always_offload_from_vram = False
|
|
|
|
for k, v in other_args.items():
|
|
setattr(args, k, v)
|
|
|
|
import ldm_patched.modules.model_management as model_management
|
|
import torch
|
|
|
|
device = model_management.get_torch_device()
|
|
torch.zeros((1, 1)).to(device, torch.float32)
|
|
model_management.soft_empty_cache()
|
|
return
|