dearwolf/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-20 18:45:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 21

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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