fix: run cloud command the second time

bentoml · Dec 19, 2024 · 08d574b · 08d574b
1 parent cd2f54d
commit 08d574b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/openllm/cloud.py b/src/openllm/cloud.py
@@ -63,6 +63,9 @@ def _get_deploy_cmd(bento: BentoInfo, target: typing.Optional[DeploymentTarget]
     base_config = resolve_cloud_config()
     if not base_config.exists():
         raise Exception('Cannot find cloud config.')
+    # remove before copy
+    if (bento.repo.path / 'bentoml' / '.yatai.yaml').exists():
+        (bento.repo.path / 'bentoml' / '.yatai.yaml').unlink()
     shutil.copy(base_config, bento.repo.path / 'bentoml' / '.yatai.yaml')
 
     return cmd, env, None