Skip to content

Commit

Permalink
Add a launcher option for notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
chiragjn committed Jan 17, 2024
1 parent e81035c commit 9a9af2c
Show file tree
Hide file tree
Showing 4 changed files with 29 additions and 3 deletions.
8 changes: 8 additions & 0 deletions .jp_app_launcher/config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
- title: Finetuning Notebook
description: Launch Truefoundry LLM Finetuning Notebook
icon: /home/jovyan/llm-finetune/assets/truefoundry-logo.svg
source: /llm-finetune/finetune.ipynb
type: notebook
cwd: /home/jovyan/llm-finetune/
catalog: Notebook
copy: false
6 changes: 3 additions & 3 deletions Dockerfile-notebook
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,9 @@ RUN sudo add-apt-repository ppa:flexiondotorg/nvtop -y && \
apt update && \
apt install -y --no-install-recommends cuda-toolkit-11-8
USER jovyan

COPY requirements.txt post-pytorch-requirements.txt /tmp_home/jovyan/llm-finetune/
COPY requirements.txt notebook-requirements.txt post-pytorch-requirements.txt /tmp_home/jovyan/llm-finetune/
RUN pip install -U pip wheel setuptools && \
pip install --no-cache-dir -U -r /tmp_home/jovyan/llm-finetune/requirements.txt jupyter_server_proxy==4.1.0 && \
pip install --no-cache-dir -U -r /tmp_home/jovyan/llm-finetune/notebook-requirements.txt && \
pip install --no-cache-dir --no-build-isolation -U -r /tmp_home/jovyan/llm-finetune/post-pytorch-requirements.txt
ENV JUPYTER_APP_LAUNCHER_PATH=/home/jovyan/llm-finetune/.jp_app_launcher/
COPY . /tmp_home/jovyan/llm-finetune/
15 changes: 15 additions & 0 deletions assets/truefoundry-logo.svg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
3 changes: 3 additions & 0 deletions notebook-requirements.txt
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
-r requirements.txt
jupyter-server-proxy==4.1.0
jupyter_app_launcher @ git+https://github.com/truefoundry/jupyter_app_launcher@9a959b894542995fc763ed07324bbd274e96610d

0 comments on commit 9a9af2c

Please sign in to comment.