diff --git a/serving/sf2f/models/model_setup.py b/serving/sf2f/models/model_setup.py
index 94ef21a..872f8f3 100644
--- 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