blob: 606633b675b89d4fb031559ccc521b18d9ca250a (
plain) (
blame)
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
|
[tool.poetry]
name = "swr2_asr"
version = "0.1.0"
description = "Automatic speech recognition with pytorch for SWR2"
authors = ["Philipp Merkel <philippmerkel@outlook.com>", "Marvin Borner <git@marvinborner.de>", "Valentin Schmidt <>", "Silja Kasper <>"]
license = "MIT"
readme = "readme.md"
packages = [{include = "swr2_asr"}]
[tool.poetry.dependencies]
python = "^3.10"
numpy = "^1.25.2"
click = "^8.1.6"
audioloader = {git = "https://github.com/KinWaiCheuk/AudioLoader.git"}
tqdm = "^4.66.1"
torch = {version = "^2.0.1+cpu", source = "pytorch-cpu"}
torchaudio = {version = "^2.0.2+cpu", source = "pytorch-cpu"}
[tool.poetry.group.dev.dependencies]
black = "^23.7.0"
pylint = "^2.17.5"
mypy = "^1.5.1"
pre-commit = "^3.3.3"
[[tool.poetry.source]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"
[[tool.poetry.source]]
name = "pytorch-gpu"
url = "https://download.pytorch.org/whl/cu118"
priority = "explicit"
[tool.poetry.scripts]
train = "swr2_asr.train:main"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
|
