From 7dcbba04f10a0697396512f47bab5ea09042788a Mon Sep 17 00:00:00 2001
From: soyoonjeong <76814748+soyoonjeong@users.noreply.github.com>
Date: Thu, 14 Mar 2024 03:26:47 +0900
Subject: [PATCH] Feat : utils import error

---
 serving/sf2f/models/model_setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/serving/sf2f/models/model_setup.py b/serving/sf2f/models/model_setup.py
index e077e47..8f31612 100644
--- a/serving/sf2f/models/model_setup.py
+++ b/serving/sf2f/models/model_setup.py
@@ -4,7 +4,10 @@
 except:
     from ..models import *
 from copy import deepcopy
-from utils import update_values, load_model_state
+try:
+    from ..utils import update_values, load_model_state
+except:
+    from .utils import update_values, load_model_state
 import glog as log
 
 def build_model(opts, image_size, checkpoint_start_from=None):