Fix : models import error4

Make-Zenerator · Mar 13, 2024 · cf7f663 · cf7f663
1 parent 5d9e8fc
commit cf7f663
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
@@ -1,5 +1,8 @@
 import torch
-import models
+try:
+    from .models import *
+except:
+    from models import *
 from copy import deepcopy
 from utils import update_values, load_model_state
 import glog as log