Fix hf_opt_multinode_jax test dependencies

triton-inference-server · Nov 17, 2023 · 8448f98 · 8448f98
1 parent 41133b1
commit 8448f98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/huggingface_opt_multinode_jax/install.sh b/examples/huggingface_opt_multinode_jax/install.sh
@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/usr/bin/env bash
 # Copyright (c) 2022-2023, NVIDIA CORPORATION. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,4 +16,4 @@ set -xe
 
 # Use 0.4.14 raises error
 pip install --upgrade "jax[cuda12_pip]!=0.4.14" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
-pip install --upgrade flax omegaconf sacrebleu SentencePiece tokenizers transformers>=4.26
+pip install --upgrade flax omegaconf sacrebleu SentencePiece tokenizers "transformers>=4.26"