ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-06-12 20:18:11 +00:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

fix(transformers*): add sentence-transformers and transformers-musicgen tests, fix musicgen wrapper (#1420)

Browse Source 
* tests: add sentence-transformers and transformers-musicgen

Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com>

* fix: tranformers-musicgen conda env

Initialize correctly the environment for the transformers-musicgen backend.

* fix(tests): transformer-musicgen tests fixups

---------

Signed-off-by: Ettore Di Giacinto <mudler@users.noreply.github.com>
This commit is contained in:
Ettore Di Giacinto
2023-12-11 13:26:02 -05:00
committed by GitHub
parent 9222bec8b1
commit 718a5d4a9e
5 changed files with 65 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/python/transformers-musicgen/test.sh
View File

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