Feat : utils import error

Make-Zenerator · Mar 13, 2024 · 7dcbba0 · 7dcbba0
1 parent 640598e
commit 7dcbba0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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):