| [tool.poetry] | |
| name = "lunar2" | |
| version = "0.1.0" | |
| description = "Training a model on Python 3.10 for HF Hub" | |
| authors = ["Lois Fernandez <loisfernandezcmp@gmail.com>"] | |
| [tool.poetry.dependencies] | |
| # Restricting to Python 3.10 specifically | |
| python = "~3.10" | |
| # Core ML stack | |
| torch = "^2.0.0" | |
| transformers = {extras = ["torch"], version = "^4.35.0"} | |
| datasets = "^2.15.0" | |
| accelerate = "^0.24.0" | |
| # For uploading to the Hugging Face Hub | |
| huggingface-hub = "^0.19.0" | |
| gymnasium = {version = "0.28.1", extras = ["box2d"]} | |
| [build-system] | |
| requires = ["poetry-core>=1.0.0"] | |
| build-backend = "poetry.core.masonry.api" |