Patryk Ptasiński commited on
Commit
3f9eaee
·
1 Parent(s): 0b5e578

Fix Stella models to use trust_remote_code=True

Browse files

- dunzhang/stella_en_400M_v5 requires custom code execution
- dunzhang/stella_en_1.5B_v5 requires custom code execution
- infgrad/stella-base-en-v2 requires custom code execution

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -27,9 +27,9 @@ MODELS = {
27
  "Qwen/Qwen3-Embedding-0.6B": {"trust_remote_code": False},
28
  "Qwen/Qwen3-Embedding-4B": {"trust_remote_code": False},
29
  "Qwen/Qwen3-Embedding-8B": {"trust_remote_code": False},
30
- "dunzhang/stella_en_400M_v5": {"trust_remote_code": False},
31
- "dunzhang/stella_en_1.5B_v5": {"trust_remote_code": False},
32
- "infgrad/stella-base-en-v2": {"trust_remote_code": False},
33
  "nvidia/NV-Embed-v2": {"trust_remote_code": True},
34
  "Alibaba-NLP/gte-Qwen2-7B-instruct": {"trust_remote_code": False},
35
  "Alibaba-NLP/gte-Qwen2-1.5B-instruct": {"trust_remote_code": False},
 
27
  "Qwen/Qwen3-Embedding-0.6B": {"trust_remote_code": False},
28
  "Qwen/Qwen3-Embedding-4B": {"trust_remote_code": False},
29
  "Qwen/Qwen3-Embedding-8B": {"trust_remote_code": False},
30
+ "dunzhang/stella_en_400M_v5": {"trust_remote_code": True},
31
+ "dunzhang/stella_en_1.5B_v5": {"trust_remote_code": True},
32
+ "infgrad/stella-base-en-v2": {"trust_remote_code": True},
33
  "nvidia/NV-Embed-v2": {"trust_remote_code": True},
34
  "Alibaba-NLP/gte-Qwen2-7B-instruct": {"trust_remote_code": False},
35
  "Alibaba-NLP/gte-Qwen2-1.5B-instruct": {"trust_remote_code": False},