From abba81d8a6172e356e7fbcbf8e7d650388f45f2e Mon Sep 17 00:00:00 2001
From: hlky <106811348+hlky@users.noreply.github.com>
Date: Mon, 15 May 2023 19:24:00 +0100
Subject: [PATCH] fix import

---
 comfy/aitemplate/model.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfy/aitemplate/model.py b/comfy/aitemplate/model.py
index b94ca2699..8aa66476d 100644
--- a/comfy/aitemplate/model.py
+++ b/comfy/aitemplate/model.py
@@ -23,9 +23,9 @@ from typing import Callable, Dict, List, NamedTuple, Optional, Tuple, TypeVar, U
 
 import numpy as np
 
-from dtype import dtype_str_to_enum
-from misc import is_linux, is_windows
-from torch_utils import torch_dtype_to_string
+from .dtype import dtype_str_to_enum
+from .misc import is_linux, is_windows
+from .torch_utils import torch_dtype_to_string
 
 # Controls how many runtimes will be used in ModelContainer by default.
 # See the runtime README.md for more information on the Model/ModelContainer