diff --git a/.gitignore b/.gitignore index baaaa2b..ecd3635 100644 --- a/.gitignore +++ b/.gitignore @@ -1,6 +1,6 @@ __pycache__ .vscode -*.txt + *.npy temp lightning_logs diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..92e1a0e --- /dev/null +++ b/requirements.txt @@ -0,0 +1,8 @@ +pytorch_lightning +einops +safetensors +transformers +plotly_express +opencv-python +tiktoken +datasets \ No newline at end of file