Dockerfile

# Базовий образ з Python
FROM python:3.10-slim

RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"

# Становлюємо env змінну для кешу
RUN --chown=user mkdir -p /cache
RUN chmod -R 777 /cache

# Становлюємо папку для result
RUN --chown=user mkdir -p /results
RUN chmod -R 777 /results

# Становлюємо папку для сбереження моделі
RUN --chown=user mkdir -p /fine_tuned_model
RUN chmod -R 777 /fine_tuned_model

ENV TRANSFORMERS_CACHE=/cache

# Оновлення 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"]