diff --git a/modelscope/pipelines/cv/ocr_detection_pipeline.py b/modelscope/pipelines/cv/ocr_detection_pipeline.py
index b54ad96d..62248714 100644
--- a/modelscope/pipelines/cv/ocr_detection_pipeline.py
+++ b/modelscope/pipelines/cv/ocr_detection_pipeline.py
@@ -17,6 +17,11 @@ from .ocr_utils import (SegLinkDetector, cal_width, combine_segments_python,
                         decode_segments_links_python, nms_python,
                         rboxes_to_polygons)
 
+if tf.__version__ >= '2.0':
+    import tf_slim as slim
+else:
+    from tensorflow.contrib import slim
+
 if tf.__version__ >= '2.0':
     tf = tf.compat.v1
 tf.compat.v1.disable_eager_execution()
diff --git a/modelscope/pipelines/cv/ocr_utils/ops.py b/modelscope/pipelines/cv/ocr_utils/ops.py
index 2bc8a8bf..eeab36a0 100644
--- a/modelscope/pipelines/cv/ocr_utils/ops.py
+++ b/modelscope/pipelines/cv/ocr_utils/ops.py
@@ -88,7 +88,7 @@ def _nn_variable(name, shape, init_method, collection=None, **kwargs):
     else:
         raise 'Unsupported weight initialization method: ' + init_method
 
-    var = tf.get_variable(name, shape=shape, initializer=initializer, **kwargs)
+    var = tf.get_variable(name, shape=shape, initializer=initializer)
     if collection is not None:
         tf.add_to_collection(collection, var)