anbench commited on
Commit
71de3dc
·
verified ·
1 Parent(s): 3315846

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. Dockerfile +12 -2
  2. requirements.txt +0 -1
Dockerfile CHANGED
@@ -7,15 +7,25 @@ ENV DEBIAN_FRONTEND=noninteractive
7
  # Install Git and build dependencies
8
  RUN apt-get update && apt-get install -y git ninja-build && apt-get clean
9
 
 
 
 
 
 
 
10
  # Update pip
11
  RUN pip install --upgrade pip
12
 
13
  # Install requirements
14
- COPY requirements.txt .
15
  RUN pip install --no-cache-dir -r requirements.txt
16
 
 
 
 
 
17
  # Copy training script
18
- COPY launch_nemotron_opus_distillation.py train.py
19
 
20
  # Command to run
21
  CMD ["python", "train.py"]
 
7
  # Install Git and build dependencies
8
  RUN apt-get update && apt-get install -y git ninja-build && apt-get clean
9
 
10
+ # Create a user with UID 1000 to avoid getpwuid() error
11
+ RUN useradd -m -u 1000 user
12
+ USER user
13
+ ENV PATH="/home/user/.local/bin:/Users/an/.git-ai/bin:/Users/an/.local/bin:/Users/an/bin:/Users/an/go/bin:/Users/an/.opencode/bin:/Users/an/.elan/bin:/Users/an/.local/bin:/opt/homebrew/bin:/opt/homebrew/sbin:/usr/local/bin:/System/Cryptexes/App/usr/bin:/usr/bin:/bin:/usr/sbin:/sbin:/var/run/com.apple.security.cryptexd/codex.system/bootstrap/usr/local/bin:/var/run/com.apple.security.cryptexd/codex.system/bootstrap/usr/bin:/var/run/com.apple.security.cryptexd/codex.system/bootstrap/usr/appleinternal/bin:/opt/pmk/env/global/bin:/Library/Apple/usr/bin:/Users/an/.cargo/bin:/Applications/iTerm.app/Contents/Resources/utilities:/Users/an/Library/Python/3.9/bin"
14
+ WORKDIR /home/user/app
15
+
16
  # Update pip
17
  RUN pip install --upgrade pip
18
 
19
  # Install requirements
20
+ COPY --chown=user requirements.txt .
21
  RUN pip install --no-cache-dir -r requirements.txt
22
 
23
+ # Explicitly install unsloth and its requirements in the correct order
24
+ RUN pip install --no-cache-dir "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git"
25
+ RUN pip install --no-cache-dir unsloth_zoo
26
+
27
  # Copy training script
28
+ COPY --chown=user launch_nemotron_opus_distillation.py train.py
29
 
30
  # Command to run
31
  CMD ["python", "train.py"]
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
 
2
- unsloth
3
  huggingface_hub[cli]
4
  trl<0.9.0
5
  peft
 
1
 
 
2
  huggingface_hub[cli]
3
  trl<0.9.0
4
  peft