Temp comment out hugginface secret

modal-labs · Feb 5, 2024 · e16ed1e · e16ed1e
1 parent 255b2ec
commit e16ed1e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 10 deletions.
diff --git a/.github/workflows/ci-cd.yml b/.github/workflows/ci-cd.yml
@@ -5,7 +5,7 @@ on: pull_request
 jobs:
   test:
     environment: CI
-    name: Deploy
+    name: Test
     runs-on: ubuntu-latest
     strategy:
       matrix:

diff --git a/src/common.py b/src/common.py
@@ -24,17 +24,16 @@
     .env(dict(HUGGINGFACE_HUB_CACHE="/pretrained", HF_HUB_ENABLE_HF_TRANSFER="1"))
 )
 
-vllm_image = (
-    Image.from_registry("nvidia/cuda:12.1.0-base-ubuntu22.04", add_python="3.10")
-    .pip_install(
-        "vllm==0.2.5",
-        "torch==2.1.2",
-        "torchvision==0.16.2",
-        "torchaudio==2.1.2"
-        )
+vllm_image = Image.from_registry(
+    "nvidia/cuda:12.1.0-base-ubuntu22.04", add_python="3.10"
+).pip_install(
+    "vllm==0.2.5",
+    "torch==2.1.2",
+    "torchvision==0.16.2",
+    "torchaudio==2.1.2",
 )
 
-stub = Stub(APP_NAME, secrets=[Secret.from_name("huggingface")])
+stub = Stub(APP_NAME)  # , secrets=[Secret.from_name("huggingface")])
 
 # Volumes for pre-trained models and training runs.
 pretrained_volume = Volume.persisted("example-pretrained-vol")