diff --git a/examples/robobrain/conf/compress.yaml b/examples/robobrain/conf/compress.yaml index 6a6bfa68a..ec0b8c6c2 100644 --- a/examples/robobrain/conf/compress.yaml +++ b/examples/robobrain/conf/compress.yaml @@ -10,6 +10,8 @@ experiment: entrypoint: flagscale/compress/compressor.py runner: hostfile: null + deploy: + use_fs_serve: true cmds: before_start: source activate flagscale-inference envs: diff --git a/flagscale/runner/runner_base.py b/flagscale/runner/runner_base.py index f1e526684..dc4dfd607 100644 --- a/flagscale/runner/runner_base.py +++ b/flagscale/runner/runner_base.py @@ -49,8 +49,10 @@ def __init__(self, config: DictConfig): backend_type = f"native_{self.task_type}" if self.config.experiment.runner.get( "deploy", None - ) is None or self.config.experiment.runner.deploy.get("use_fs_serve", False): - raise ValueError("config.experiment.deploy.use_fs_serve in YAML should be true") + ) is None or not self.config.experiment.runner.deploy.get("use_fs_serve", False): + raise ValueError( + "when backend type is native, config.experiment.deploy.use_fs_serve in YAML should be true" + ) self.backend_type = backend_type