Update initialization.py

initialization

initialization

Update initialization.py

modules/initialize.py

@@ -12,6 +12,10 @@ def imports():
  logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR) # sshh...
  logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
 
+ from modules_forge.initialization import initialize_forge
+ initialize_forge()
+ startup_timer.record("initialize forge")
+
  import torch # noqa: F401
  startup_timer.record("import torch")
  import pytorch_lightning # noqa: F401

modules_forge/initialization.py

@@ -0,0 +1,14 @@
+
+
+def initialize_forge():
+ import ldm_patched.modules.args_parser as args_parser
+
+ args_parser.args = args_parser.parser.parse_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()
+ return