: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 |
|