Create Dockerfile

Dockerfile

@@ -0,0 +1,14 @@
+# Use an official PyTorch image as the base
+FROM pytorch/pytorch:latest
+
+# Set working directory
+WORKDIR /app
+
+# Install necessary Python libraries
+RUN pip install --no-cache-dir transformers datasets accelerate peft torch torchvision torchaudio
+
+# Copy all local files into the container
+COPY . .
+
+# Set default command (replace with your training script later)
+CMD ["python", "train.py"]