Spaces:
Paused
Paused
File size: 1,133 Bytes
9228840 12a1a3f 4724042 9228840 47711b0 5f1f151 1c31b34 47711b0 cc562a8 47711b0 4724042 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
# Базовий образ з Python
FROM python:3.10-slim
# Создаем пользователя и директорию
RUN useradd -m -u 1000 user
# Создаем директорию для кеша и устанавливаем права
RUN mkdir -p /cache && chown -R user:user /cache
RUN mkdir -p /results && chown -R user:user /results
RUN mkdir -p /fine_tuned_model && chown -R user:user /fine_tuned_model
# Устанавливаем переменную окружения для кеша
ENV TRANSFORMERS_CACHE=/cache
# Переключаемся на созданного пользователя
USER user
ENV PATH="/home/user/.local/bin:$PATH"
# Оновлення pip та встановлення потрібних бібліотек
RUN pip install --upgrade pip
# Встановлення необхідних пакетів
RUN pip install torch transformers datasets tqdm
RUN pip install transformers[torch]
# Скопіювати ваш Python-код у контейнер
WORKDIR /app
COPY --chown=user . /app
# Запуск скрипта донавчання
CMD ["python3", "finetunning.py"] |