36 lines
787 B
TOML
36 lines
787 B
TOML
[project]
|
|
name = "codebert-training"
|
|
version = "1.0.0"
|
|
description = "This code is used to train CodeBERT"
|
|
authors = [
|
|
{name = "Patryk Bartkowiak", email = "patbar15@st.amu.edu.pl"},
|
|
]
|
|
dependencies = [
|
|
"wandb==0.18.5",
|
|
"torch==2.5.0",
|
|
"tqdm==4.66.5",
|
|
"tree-sitter==0.23.1",
|
|
"transformers==4.45.2",
|
|
"datasets==3.0.1",
|
|
"huggingface-hub==0.26.0",
|
|
]
|
|
requires-python = "==3.11.*"
|
|
readme = "README.md"
|
|
license = {text = "MIT"}
|
|
|
|
[[tool.pdm.source]]
|
|
url = "https://download.pytorch.org/whl/cu124"
|
|
verify_ssl = true
|
|
include_packages = ["torch", "torchvision", "torchaudio"]
|
|
name = "cuda"
|
|
|
|
[build-system]
|
|
requires = ["pdm-backend"]
|
|
build-backend = "pdm.backend"
|
|
|
|
[tool.pdm]
|
|
distribution = true
|
|
|
|
[tool.pdm.scripts]
|
|
run_training = {cmd = "src/train_codebert_mlm.py"}
|