FROM ubuntu:22.04 AS builder

# Установка зависимостей

RUN apt update && apt-get install -y \
    python3 pip git curl wget nano sudo apt-utils && \
    rm -rf /var/lib/apt/lists/*

# Установка RKLLM Toolkit
RUN git clone https://github.com/airockchip/rknn-llm

# Установка Python-пакета
RUN pip3 install /ezrknn-llm/rkllm-toolkit/packages/rkllm_toolkit-1.2.3-cp312-cp312-linux_x86_64.whl

# Клонирование модели
WORKDIR /models

# Копируем скрипт компиляции
COPY entrypoint.py /entrypoint.py

# Точка входа для компиляции
ENTRYPOINT ["python3", "/entrypoint.py"]