fix: hot fix for gemma2 locks

bentoml · Aug 12, 2024 · d405a5b · d405a5b
1 parent 0712e5e
commit d405a5b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/bentoml/bentos/gemma2/27b-instruct-fp16-9799/env/python/requirements.lock.txt b/bentoml/bentos/gemma2/27b-instruct-fp16-9799/env/python/requirements.lock.txt
@@ -1,3 +1,4 @@
+--extra-index-url https://flashinfer.ai/whl/cu121/torch2.3
 aiohappyeyeballs==2.3.5
 aiohttp==3.10.3
 aiosignal==1.3.1

diff --git a/bentoml/bentos/gemma2/9b-instruct-fp16-cb2b/env/python/requirements.lock.txt b/bentoml/bentos/gemma2/9b-instruct-fp16-cb2b/env/python/requirements.lock.txt
@@ -1,3 +1,4 @@
+--extra-index-url https://flashinfer.ai/whl/cu121/torch2.3
 aiohappyeyeballs==2.3.5
 aiohttp==3.10.3
 aiosignal==1.3.1