raghavNCI
commited on
Commit
·
02e2d96
1
Parent(s):
e30a3df
env names correction
Browse files
models_initialization/mistral_registry.py
CHANGED
|
@@ -3,8 +3,8 @@ import json
|
|
| 3 |
import requests
|
| 4 |
from requests.exceptions import RequestException
|
| 5 |
|
| 6 |
-
HF_ENDPOINT_URL = os.getenv("
|
| 7 |
-
HF_ENDPOINT_TOKEN = os.getenv("
|
| 8 |
|
| 9 |
assert HF_ENDPOINT_URL, "❌ HF_ENDPOINT_URL is not set"
|
| 10 |
assert HF_ENDPOINT_TOKEN, "❌ HF_ENDPOINT_TOKEN is not set"
|
|
|
|
| 3 |
import requests
|
| 4 |
from requests.exceptions import RequestException
|
| 5 |
|
| 6 |
+
HF_ENDPOINT_URL = os.getenv("SAGEMAKER_ENDPOINT_NAME")
|
| 7 |
+
HF_ENDPOINT_TOKEN = os.getenv("HF_TOKEN")
|
| 8 |
|
| 9 |
assert HF_ENDPOINT_URL, "❌ HF_ENDPOINT_URL is not set"
|
| 10 |
assert HF_ENDPOINT_TOKEN, "❌ HF_ENDPOINT_TOKEN is not set"
|