Spaces:

mindchain
/

rlm-arithmetic-training

Runtime error

Upload Dockerfile with huggingface_hub

440ca06 verified 8 months ago

587 Bytes

FROM python:3.11-slim

WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    && rm -rf /var/lib/apt/lists/*

# Install core Python packages ONLY - no IPEX, no OpenVINO
RUN pip install --no-cache-dir \
    torch \
    transformers \
    accelerate \
    trl \
    datasets \
    huggingface_hub \
    bitsandbytes

# Note: NO peft - we're testing without LoRA!

# Copy training script
COPY train_arithmetic_v9_no_lora.py .

# Set HF token from secret
ENV HF_TOKEN=${HF_TOKEN}

# Run training
CMD ["python", "train_arithmetic_v9_no_lora.py"]

	FROM python:3.11-slim

	WORKDIR /app

	# Install system dependencies
	RUN apt-get update && apt-get install -y \
	git \
	&& rm -rf /var/lib/apt/lists/*

	# Install core Python packages ONLY - no IPEX, no OpenVINO
	RUN pip install --no-cache-dir \
	torch \
	transformers \
	accelerate \
	trl \
	datasets \
	huggingface_hub \
	bitsandbytes

	# Note: NO peft - we're testing without LoRA!

	# Copy training script
	COPY train_arithmetic_v9_no_lora.py .

	# Set HF token from secret
	ENV HF_TOKEN=${HF_TOKEN}

	# Run training
	CMD ["python", "train_arithmetic_v9_no_lora.py"]