Fix compatibility issues

f743bf9a · haoyuying · GitHub · cf9f0cbb · f743bf9a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddlehub/module/cv_module.py paddlehub/module/cv_module.py +1 -1

未找到文件。
--- a/paddlehub/module/cv_module.py
+++ b/paddlehub/module/cv_module.py
@@ -193,7 +193,7 @@ class ImageColorizeModule(RunModule, ImageServing):
        Returns:
            results(dict) : The model outputs, such as metrics.
        '''
-        if Version(paddle.__version__) >= '2.1':
+        if Version(paddle.__version__) >= '2.1' or Version(paddle.__version__) == '0.0.0':
            img = self.preprocess(batch)
        else:
            img = self.preprocess(batch[0])