Spaces:
Sleeping
Sleeping
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]
|