Fix : models import error2

Make-Zenerator · Mar 13, 2024 · 3e95a90 · 3e95a90
1 parent a6b3baa
commit 3e95a90
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/serving/sf2f/inference.py b/serving/sf2f/inference.py
@@ -1,4 +1,7 @@
-import .models
+try:
+    from . import models
+except:
+    import .models as models
 import torch
 import os, glob
 import mlflow