diff --git a/comfy/model_management_types.py b/comfy/model_management_types.py
index 41add063e..25c8f917c 100644
--- a/comfy/model_management_types.py
+++ b/comfy/model_management_types.py
@@ -7,11 +7,12 @@ import torch
 import torch.nn
 from typing_extensions import TypedDict, NotRequired
 
-from comfy.latent_formats import LatentFormat
+from .latent_formats import LatentFormat
 
 ModelManageableT = TypeVar('ModelManageableT', bound='ModelManageable')
 LatentFormatT = TypeVar('LatentFormatT', bound=LatentFormat)
 
+
 @runtime_checkable
 class DeviceSettable(Protocol):
     @property