:boom: [Fix] Missing HF_ENDPOINT config in hf space
Browse files- configs/info.json +1 -1
- transforms/embed.py +2 -1
configs/info.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"version": "0.1.
|
3 |
"app_name": "Embedding API",
|
4 |
"server": "0.0.0.0",
|
5 |
"port": 16666
|
|
|
1 |
{
|
2 |
+
"version": "0.1.3",
|
3 |
"app_name": "Embedding API",
|
4 |
"server": "0.0.0.0",
|
5 |
"port": 16666
|
transforms/embed.py
CHANGED
@@ -9,7 +9,8 @@ from numpy.linalg import norm
|
|
9 |
from configs.envs import ENVS
|
10 |
from configs.constants import AVAILABLE_MODELS
|
11 |
|
12 |
-
|
|
|
13 |
os.environ["HF_TOKEN"] = ENVS["HF_TOKEN"]
|
14 |
|
15 |
|
|
|
9 |
from configs.envs import ENVS
|
10 |
from configs.constants import AVAILABLE_MODELS
|
11 |
|
12 |
+
if ENVS["HF_ENDPOINT"]:
|
13 |
+
os.environ["HF_ENDPOINT"] = ENVS["HF_ENDPOINT"]
|
14 |
os.environ["HF_TOKEN"] = ENVS["HF_TOKEN"]
|
15 |
|
16 |
|