diff --git a/deepspeed/inference/engine.py b/deepspeed/inference/engine.py index 6f9d90e3f743..d51cfda79001 100755 --- a/deepspeed/inference/engine.py +++ b/deepspeed/inference/engine.py @@ -420,7 +420,7 @@ def _load_checkpoint(self, load_dir, load_module_strict=True, tag=None): if is_pipe_parallel: raise RuntimeError( 'pipeline parallelism is currently not supported in inference.') - if os.path.isdir(load_dir): + if not isinstance(load_dir, dict) and os.path.isdir(load_dir): if tag is None: latest_path = os.path.join(load_dir, "latest") if os.path.isfile(latest_path):