Skip to content

Commit d80cc96

Browse files
mickr777psychedelicious
authored andcommitted
Delay Imports that require torch
1 parent 7ccabfa commit d80cc96

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

invokeai/app/run_app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,6 @@ def run_app() -> None:
1717

1818
import uvicorn
1919

20-
from invokeai.app.invocations.baseinvocation import InvocationRegistry
21-
from invokeai.app.invocations.load_custom_nodes import load_custom_nodes
2220
from invokeai.app.services.config.config_default import get_config
2321
from invokeai.app.util.torch_cuda_allocator import configure_torch_cuda_allocator
2422
from invokeai.backend.util.logging import InvokeAILogger
@@ -35,6 +33,8 @@ def run_app() -> None:
3533

3634
# This import must happen after configure_torch_cuda_allocator() is called, because the module imports torch.
3735
from invokeai.backend.util.devices import TorchDevice
36+
from invokeai.app.invocations.baseinvocation import InvocationRegistry
37+
from invokeai.app.invocations.load_custom_nodes import load_custom_nodes
3838

3939
torch_device_name = TorchDevice.get_torch_device_name()
4040
logger.info(f"Using torch device: {torch_device_name}")

0 commit comments

Comments
 (0)