Spaces:
Running
Running
Commit
Β·
add2495
1
Parent(s):
3bc689a
modified files
Browse files- Dockerfile +2 -6
- app.py +8 -10
Dockerfile
CHANGED
@@ -2,15 +2,11 @@
|
|
2 |
# you will also find guides on how best to write your Dockerfile
|
3 |
|
4 |
FROM python:3.9
|
5 |
-
|
6 |
-
echo "Using ClearML API key: $(cat /run/secrets/CLEARML_API_ACCESS_KEY)" && \
|
7 |
-
pip install clearml
|
8 |
RUN useradd -m -u 1000 user
|
9 |
USER user
|
10 |
ENV PATH="/home/user/.local/bin:$PATH"
|
11 |
-
|
12 |
-
ENV CLEARML_API_HOST="https://api.clear.ml"
|
13 |
-
ENV CLEARML_FILES_HOST="https://files.clear.ml"
|
14 |
WORKDIR /app
|
15 |
|
16 |
COPY --chown=user ./requirements.txt requirements.txt
|
|
|
2 |
# you will also find guides on how best to write your Dockerfile
|
3 |
|
4 |
FROM python:3.9
|
5 |
+
|
|
|
|
|
6 |
RUN useradd -m -u 1000 user
|
7 |
USER user
|
8 |
ENV PATH="/home/user/.local/bin:$PATH"
|
9 |
+
|
|
|
|
|
10 |
WORKDIR /app
|
11 |
|
12 |
COPY --chown=user ./requirements.txt requirements.txt
|
app.py
CHANGED
@@ -11,24 +11,22 @@ import argparse
|
|
11 |
from argparse import Namespace
|
12 |
from peft import PeftModel, PeftConfig, get_peft_model, LoraConfig
|
13 |
|
14 |
-
#
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
CLEARML_ACCESS_KEY = os.environ["CLEARML_API_ACCESS_KEY"]
|
19 |
CLEARML_SECRET_KEY = os.environ["CLEARML_API_SECRET_KEY"]
|
20 |
-
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|
21 |
|
22 |
-
#
|
23 |
Task.set_credentials(
|
24 |
-
api_host=
|
25 |
-
web_host=
|
26 |
-
files_host=
|
27 |
key=CLEARML_ACCESS_KEY,
|
28 |
secret=CLEARML_SECRET_KEY,
|
29 |
)
|
30 |
|
31 |
-
MAX_SOURCE_LENGTH = 512
|
32 |
|
33 |
def pad_assert(tokenizer, source_ids):
|
34 |
source_ids = source_ids[:MAX_SOURCE_LENGTH - 2]
|
|
|
11 |
from argparse import Namespace
|
12 |
from peft import PeftModel, PeftConfig, get_peft_model, LoraConfig
|
13 |
|
14 |
+
# Load endpoints & creds
|
15 |
+
CLEARML_API_HOST = os.environ["CLEARML_API_HOST"]
|
16 |
+
CLEARML_WEB_HOST = os.environ["CLEARML_WEB_HOST"]
|
17 |
+
CLEARML_FILES_HOST = os.environ["CLEARML_FILES_HOST"]
|
18 |
CLEARML_ACCESS_KEY = os.environ["CLEARML_API_ACCESS_KEY"]
|
19 |
CLEARML_SECRET_KEY = os.environ["CLEARML_API_SECRET_KEY"]
|
|
|
20 |
|
21 |
+
# Apply to SDK
|
22 |
Task.set_credentials(
|
23 |
+
api_host=CLEARML_API_HOST,
|
24 |
+
web_host=CLEARML_WEB_HOST,
|
25 |
+
files_host=CLEARML_FILES_HOST,
|
26 |
key=CLEARML_ACCESS_KEY,
|
27 |
secret=CLEARML_SECRET_KEY,
|
28 |
)
|
29 |
|
|
|
30 |
|
31 |
def pad_assert(tokenizer, source_ids):
|
32 |
source_ids = source_ids[:MAX_SOURCE_LENGTH - 2]
|