my-sd/modules_forge/initialization.py

28 lines
788 B
Python
Raw Normal View History

def initialize_forge():
2024-01-27 19:43:54 +00:00
from ldm_patched.modules import args_parser
2024-01-27 19:38:37 +00:00
args_parser.args, _ = args_parser.parser.parse_known_args()
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()
2024-01-24 20:25:32 +00:00
import modules_forge.patch_clip
modules_forge.patch_clip.patch_all_clip()
import modules_forge.patch_precision
modules_forge.patch_precision.patch_all_precision()
2024-01-28 04:42:38 +00:00
import modules_forge.patch_basic
modules_forge.patch_basic.patch_all_basics()
2024-01-25 23:13:32 +00:00
if model_management.directml_enabled:
model_management.lowvram_available = True
model_management.OOM_EXCEPTION = Exception
return