Spaces:
Running on Zero
Running on Zero
Switch LLM2Vec to Meta-Llama-3.1-8B-Instruct (accessible with current token)
Browse files- app.py +1 -1
- kimodo/model/load_model.py +1 -1
- kimodo/scripts/run_text_encoder_server.py +1 -1
app.py
CHANGED
|
@@ -28,7 +28,7 @@ os.environ.setdefault("HF_MODE", "1")
|
|
| 28 |
# Avoid local LLM2Vec fallback on Spaces (requires gated Llama weights).
|
| 29 |
os.environ.setdefault("TEXT_ENCODER_MODE", "api")
|
| 30 |
os.environ.setdefault("TEXT_ENCODER", "llm2vec")
|
| 31 |
-
os.environ.setdefault("LLM2VEC_BASE_MODEL", "
|
| 32 |
os.environ.setdefault(
|
| 33 |
"LLM2VEC_PEFT_MODEL",
|
| 34 |
"McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp-supervised",
|
|
|
|
| 28 |
# Avoid local LLM2Vec fallback on Spaces (requires gated Llama weights).
|
| 29 |
os.environ.setdefault("TEXT_ENCODER_MODE", "api")
|
| 30 |
os.environ.setdefault("TEXT_ENCODER", "llm2vec")
|
| 31 |
+
os.environ.setdefault("LLM2VEC_BASE_MODEL", "meta-llama/Meta-Llama-3.1-8B-Instruct")
|
| 32 |
os.environ.setdefault(
|
| 33 |
"LLM2VEC_PEFT_MODEL",
|
| 34 |
"McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp-supervised",
|
kimodo/model/load_model.py
CHANGED
|
@@ -20,7 +20,7 @@ from .loading import (
|
|
| 20 |
from .registry import get_model_info, resolve_model_name
|
| 21 |
|
| 22 |
DEFAULT_TEXT_ENCODER = "llm2vec"
|
| 23 |
-
DEFAULT_LLM2VEC_BASE = "
|
| 24 |
DEFAULT_LLM2VEC_PEFT = "McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp-supervised"
|
| 25 |
TEXT_ENCODER_PRESETS = {
|
| 26 |
"llm2vec": {
|
|
|
|
| 20 |
from .registry import get_model_info, resolve_model_name
|
| 21 |
|
| 22 |
DEFAULT_TEXT_ENCODER = "llm2vec"
|
| 23 |
+
DEFAULT_LLM2VEC_BASE = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
| 24 |
DEFAULT_LLM2VEC_PEFT = "McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp-supervised"
|
| 25 |
TEXT_ENCODER_PRESETS = {
|
| 26 |
"llm2vec": {
|
kimodo/scripts/run_text_encoder_server.py
CHANGED
|
@@ -17,7 +17,7 @@ DEFAULT_SERVER_NAME = "0.0.0.0"
|
|
| 17 |
DEFAULT_SERVER_PORT = 9550
|
| 18 |
DEFAULT_TMP_FOLDER = "/tmp/text_encoder/"
|
| 19 |
DEFAULT_TEXT_ENCODER = "llm2vec"
|
| 20 |
-
DEFAULT_LLM2VEC_BASE = "
|
| 21 |
DEFAULT_LLM2VEC_PEFT = "McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp-supervised"
|
| 22 |
TEXT_ENCODER_PRESETS = {
|
| 23 |
"llm2vec": {
|
|
|
|
| 17 |
DEFAULT_SERVER_PORT = 9550
|
| 18 |
DEFAULT_TMP_FOLDER = "/tmp/text_encoder/"
|
| 19 |
DEFAULT_TEXT_ENCODER = "llm2vec"
|
| 20 |
+
DEFAULT_LLM2VEC_BASE = "meta-llama/Meta-Llama-3.1-8B-Instruct"
|
| 21 |
DEFAULT_LLM2VEC_PEFT = "McGill-NLP/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp-supervised"
|
| 22 |
TEXT_ENCODER_PRESETS = {
|
| 23 |
"llm2vec": {
|