From 52e5fa25ec5a890f83b32377e056779b2c3d4635 Mon Sep 17 00:00:00 2001
From: ZHO-ZHO-ZHO <140084057+ZHO-ZHO-ZHO@users.noreply.github.com>
Date: Fri, 22 Mar 2024 21:07:59 +0800
Subject: [PATCH] V1.0

---
 depthfm/unet/attention.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/depthfm/unet/attention.py b/depthfm/unet/attention.py
index 6a30ab1..f2d2982 100644
--- a/depthfm/unet/attention.py
+++ b/depthfm/unet/attention.py
@@ -6,7 +6,7 @@ from inspect import isfunction
 import torch.nn.functional as F
 from typing import Optional, Any
 
-from util import checkpoint
+from .util import checkpoint
 
 
 try: