|
|
@ -76,7 +76,7 @@ class Predictor(object):
|
|
|
|
enable_mkldnn=False):
|
|
|
|
enable_mkldnn=False):
|
|
|
|
self.batch_size = batch_size
|
|
|
|
self.batch_size = batch_size
|
|
|
|
|
|
|
|
|
|
|
|
if paddlespeech.utils.satisfy_paddle_version('2.6.0'):
|
|
|
|
if paddlespeech.utils.satisfy_paddle_version('3.0.0b'):
|
|
|
|
config = inference.Config(model_dir, 'inference')
|
|
|
|
config = inference.Config(model_dir, 'inference')
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
model_file = os.path.join(model_dir, 'inference.pdmodel')
|
|
|
|
model_file = os.path.join(model_dir, 'inference.pdmodel')
|
|
|
|