rydlrKE commited on
Commit
cc3cd7d
·
1 Parent(s): 1dbab1b

Switch LLM2Vec to Meta-Llama-3.1-8B-Instruct (accessible with current token)

Browse files
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", "rydlrKE/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp")
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 = "rydlrKE/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp"
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 = "rydlrKE/LLM2Vec-Meta-Llama-3-8B-Instruct-mntp"
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": {