optimum

Author: rebel-eunji

.github/workflows/rbln_optimum_ci.yaml

@@ -100,11 +100,15 @@ jobs:
             uv pip install --system packaging setuptools wheel simphile pynvml huggingface_hub setuptools_scm fire pytest pytest-asyncio
             VERSION=${{ inputs.optimum_rbln_version }}
             if [ -n "${{ inputs.optimum_rbln_version }}" ]; then
-              uv pip install --system --force-reinstall --no-cache-dir dist/vllm_rbln*.whl --constraint <(echo "optimum-rbln==${{ inputs.optimum_rbln_version }}") --index-url http://pypi-cache.devpi.svc.cluster.local/root/pypi/+simple/ --extra-index-url https://pypi.org/simple
-              echo "optimum-rbln ${{ inputs.optimum_rbln_version }}"
+              uv pip install --system --force-reinstall --no-cache-dir dist/vllm_rbln*.whl --index-url http://pypi-cache.devpi.svc.cluster.local/root/pypi/+simple/ --extra-index-url https://pypi.org/simple
             else
               uv pip install --system --force-reinstall --no-cache-dir dist/vllm_rbln*.whl --index-url http://pypi-cache.devpi.svc.cluster.local/root/pypi/+simple/ --extra-index-url https://pypi.org/simple
             fi
+            git clone https://github.com/huggingface/optimum-rbln.git
+            cd optimum-rbln
+            git checkout update_torch_version
+            ./scripts/uv-sync.sh
+            cd ..
 
       - name: Run gemma2 model (hybrid attention) (V1)
         run: >