Explorar el Código

Merge pull request #125 from khoags/master

Update vllm version for Colab notebook
Qing Yan hace 2 meses
padre
commit
2562ed0df4
Se han modificado 1 ficheros con 2 adiciones y 2 borrados
  1. 2 2
      demo/demo_colab_remote_server.ipynb

+ 2 - 2
demo/demo_colab_remote_server.ipynb

@@ -602,7 +602,7 @@
         "os.environ[\"PYTHONPATH\"] = f\"{os.path.dirname(hf_model_path)}:{os.environ.get('PYTHONPATH', '')}\"\n",
         "\n",
         "# Install required packages\n",
-        "!pip install vllm transformers\n",
+        "!pip install vllm==0.10.0 transformers\n",
         "\n",
         "# Modify vllm import (this is a workaround - may need adjustment based on vllm version)\n",
         "try:\n",
@@ -1163,4 +1163,4 @@
       "outputs": []
     }
   ]
-}
+}