feat: migrate transformers-musicgen to uv

Signed-off-by: Chris Jowett <421501+cryptk@users.noreply.github.com>

backend/python/transformers-musicgen/Makefile

@@ -1,6 +1,6 @@
 .PHONY: transformers-musicgen
 transformers-musicgen: protogen
-	$(MAKE) -C ../common-env/transformers
+	bash install.sh
 
 .PHONY: run
 run: protogen
@@ -22,4 +22,8 @@ protogen-clean:
 	$(RM) backend_pb2_grpc.py backend_pb2.py
 
 backend_pb2_grpc.py backend_pb2.py:
-	python3 -m grpc_tools.protoc -I../.. --python_out=. --grpc_python_out=. backend.proto
+	python3 -m grpc_tools.protoc -I../.. --python_out=. --grpc_python_out=. backend.proto
+
+.PHONY: clean
+clean: protogen-clean
+	rm -rf venv

backend/python/transformers-musicgen/install.sh

@@ -0,0 +1,25 @@
+#!/bin/bash
+set -ex
+
+MY_DIR="$(dirname -- "${BASH_SOURCE[0]}")"
+
+python -m venv ${MY_DIR}/venv
+source ${MY_DIR}/venv/bin/activate
+
+uv pip install --requirement ${MY_DIR}/requirements.txt
+
+if [ -f "requirements-${BUILD_TYPE}.txt" ]; then
+    uv pip install --requirement ${MY_DIR}/requirements-${BUILD_TYPE}.txt
+fi
+
+if [ -d "/opt/intel" ]; then
+    # Intel GPU: If the directory exists, we assume we are using the Intel image
+    # https://github.com/intel/intel-extension-for-pytorch/issues/538
+    if [ -f "requirements-intel.txt" ]; then
+        uv pip install --index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/ --requirement ${MY_DIR}/requirements-intel.txt
+    fi
+fi
+
+if [ "$PIP_CACHE_PURGE" = true ] ; then
+    pip cache purge
+fi

backend/python/transformers-musicgen/requirements.txt

@@ -0,0 +1,5 @@
+transformers>=4.38.2
+grpcio==1.63.0
+protobuf
+torch
+scipy==1.13.0

backend/python/transformers-musicgen/run.sh

@@ -1,16 +1,10 @@
 #!/bin/bash
 
 ##
-## A bash script wrapper that runs the transformers-musicgen server with conda
+## A bash script wrapper that runs the GRPC backend
 
-echo "Launching gRPC server for transformers-musicgen"
+MY_DIR="$(dirname -- "${BASH_SOURCE[0]}")"
 
-export PATH=$PATH:/opt/conda/bin
+source $MY_DIR/venv/bin/activate
 
-# Activate conda environment
-source activate transformers
-
-# get the directory where the bash script is located
-DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
-
-python $DIR/transformers_server.py $@
+python $MY_DIR/transformers_server.py $@

backend/python/transformers-musicgen/test.sh

@@ -1,11 +1,10 @@
 #!/bin/bash
+
 ##
-## A bash script wrapper that runs the transformers server with conda
+## A bash script wrapper that runs the tests
 
-# Activate conda environment
-source activate transformers
+MY_DIR="$(dirname -- "${BASH_SOURCE[0]}")"
 
-# get the directory where the bash script is located
-DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
+source $MY_DIR/venv/bin/activate
 
-python -m unittest $DIR/test_transformers.py
+python -m unittest $MY_DIR/test_transformers_server.py