Skip to content
This repository has been archived by the owner on Oct 4, 2024. It is now read-only.

Commit

Permalink
add dummy test app
Browse files Browse the repository at this point in the history
  • Loading branch information
justusschock committed Feb 8, 2023
1 parent 0066dee commit 65b0ece
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions dummy_test_app.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
#! pip install git+https://github.com/Lightning-AI/LAI-TLDR-Component git+https://github.com/Lightning-AI/lightning-LLMs
#! curl https://raw.githubusercontent.com/Shivanandroy/T5-Finetuning-PyTorch/main/data/news_summary.csv --create-dirs -o ${HOME}/data/summary/news.csv -C -

import lightning
from tests.test_app import DummyTLDR
from lit_llms.tensorboard import MultiNodeLightningTrainerWithTensorboard
app = lightning.LightningApp(
MultiNodeLightningTrainerWithTensorboard(
DummyTLDR, num_nodes=2, cloud_compute=lightning.CloudCompute("gpu-fast-multi", disk_size=50),
)
)

0 comments on commit 65b0ece

Please sign in to comment.