Feat: new backend: transformers-musicgen (#1387)

Transformers-MusicGen
---------

Signed-off-by: Dave <dave@gray101.com>
This commit is contained in:
Dave 2023-12-08 04:01:02 -05:00 committed by GitHub
parent 6011911746
commit 8b6e601405
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
26 changed files with 868 additions and 13 deletions

View file

@ -0,0 +1,25 @@
TRANSFORMERS_MUSICGEN_CONDA_PATH = "transformers-musicgen.yml"
ifeq ($(BUILD_TYPE), cublas)
TRANSFORMERS_MUSICGEN_CONDA_PATH = "transformers-musicgen-nvidia.yml"
endif
.PHONY: transformers-musicgen
transformers-musicgen:
@echo "Creating virtual environment..."
@conda env create --name transformers-musicgen --file $(TRANSFORMERS_MUSICGEN_CONDA_PATH)
@echo "Virtual environment created."
.PHONY: run
run:
@echo "Running transformers..."
bash run.sh
@echo "transformers run."
# It is not working well by using command line. It only6 works with IDE like VSCode.
.PHONY: test
test:
@echo "Testing transformers..."
bash test.sh
@echo "transformers tested."