-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
44 lines (41 loc) · 1.73 KB
/
pyproject.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"
[project]
name = "lamoto"
version = "2024.10.01" # Don't forget to change to change the lamoto.__version__ variable.
requires-python = ">= 3.10"
authors = [
{name = "Thomas Bauwens", email = "[email protected]"}
]
maintainers = [
{name = "Thomas Bauwens", email = "[email protected]"}
]
description = "Language Modelling Tasks as Objects (LaMoTO) provides a framework for language model training (masked and causal, pretraining and finetuning) where the tasks, not just the models, are classes themselves."
readme = "README.md"
license = {file = "LICENSE"}
keywords = ["language models", "LMs", "NLP", "PyTorch", "OOP", "object-oriented"]
classifiers = [
"Programming Language :: Python :: 3",
"Operating System :: OS Independent",
]
dependencies = [
"transformers >= 4.24.0",
"datasets >= 2.18.0",
"evaluate >= 0.4.1",
"pyfiglet >= 1.0.2",
"wandb",
"scikit-learn", # Technically a dependency of evaluate, but they refuse to add it to their metadata.
"conllu",
"supar @ git+https://github.com/yzhangcs/parser",
"hf_mtask_trainer @ git+https://github.com/zipzou/hf-multitask-trainer"
]
# These GitHub dependencies aren't optional, but if you already have them installed as editable, pip will override
# that installation with an installation in site-packages. Hence, if this is the case for you, install these manually.
[project.optional-dependencies]
github = [
"archit[github] @ git+https://github.com/bauwenst/ArchIt",
"tktkt[github] @ git+https://github.com/bauwenst/TkTkT",
"modest[github] @ git+https://github.com/bauwenst/MoDeST",
"fiject @ git+https://github.com/bauwenst/fiject"
]