michaelfeil commited on
Commit
40501a3
·
verified ·
1 Parent(s): 5fae804

Update start-infinity-emb.sh

Browse files
Files changed (1) hide show
  1. start-infinity-emb.sh +1 -1
start-infinity-emb.sh CHANGED
@@ -1,4 +1,4 @@
1
  export HF_HOME=/tmp/cache
2
  export TRANSFORMERS_CACHE=/tmp/cache
3
  export INFINITY_QUEUE_SIZE=512
4
- infinity_emb v2 --model-id michaelfeil/bge-small-en-v1.5 --model-id BAAI/bge-reranker-base --port 7860 --batch-size 8 --engine torch --lengths-via-tokenize
 
1
  export HF_HOME=/tmp/cache
2
  export TRANSFORMERS_CACHE=/tmp/cache
3
  export INFINITY_QUEUE_SIZE=512
4
+ infinity_emb v2 --model-id michaelfeil/bge-small-en-v1.5 --model-id BAAI/bge-reranker-base --model-id SamLowe/roberta-base-go_emotions --port 7860 --batch-size 8 --engine torch --lengths-via-tokenize