Fix TGI-Gaudi Docker Issue (#177)

Signed-off-by: letonghan <letong.han@intel.com> Signed-off-by: lvliang-intel <liang1.lv@intel.com> Signed-off-by: Yue, Wenjiao <wenjiao.yue@intel.com> Signed-off-by: chensuyue <suyue.chen@intel.com>
2024-05-25 22:53:57 +08:00
parent a6b3caf128
commit 26ae49c9dd
4 changed files with 13 additions and 4 deletions
--- a/CodeTrans/docker-composer/gaudi/docker_compose.yaml
+++ b/CodeTrans/docker-composer/gaudi/docker_compose.yaml
@@ -25,9 +25,9 @@ services:
    environment:
      http_proxy: ${http_proxy}
      https_proxy: ${https_proxy}
-      RUNTIME: habana
      HABANA_VISIBLE_DEVICES: all
      OMPI_MCA_btl_vader_single_copy_mechanism: none
+    runtime: habana
    cap_add:
      - SYS_NICE
    ipc: host