anugrah55 commited on
Commit
a72a735
·
verified ·
1 Parent(s): e544913

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +13 -0
Dockerfile ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use a base image with CUDA and Python
2
+ FROM huggingface/transformers-pytorch-gpu:latest
3
+
4
+ # Copy all the files from the repo to the container
5
+ COPY . /app
6
+ WORKDIR /app
7
+
8
+ # Install dependencies
9
+ RUN pip install -r requirements.txt
10
+ RUN pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" trl peft bitsandbytes requests
11
+
12
+ # Run the training script
13
+ CMD ["python", "hf_train_runner.py"]