-
Notifications
You must be signed in to change notification settings - Fork 8
Expand file tree
/
Copy pathenvironment.yml
More file actions
44 lines (39 loc) · 792 Bytes
/
Copy pathenvironment.yml
File metadata and controls
44 lines (39 loc) · 792 Bytes
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
name: t_ragx
channels:
- nvidia
- pytorch
- conda-forge
dependencies:
- python=3.10
- pytorch==2.1.1
- torchvision
- torchaudio
- pip
- pip:
- t-ragx
- fasttext-wheel>=0.9.2
- elasticsearch>=7.9.1
- levenshtein~=0.23.0
- llama-cpp-python~=0.2.54
- transformers~=4.38.1
- unbabel-comet>=2.2.1
- python-dotenv>=0.5.1
- openai~=1.13.3
- OpenCC
- datasets==2.13.1
- fsspec==2023.9.2
- fugashi[unidic_lite]
- jinja2~=3.1.2
- tqdm~=4.65.0
- pandas
- datasets==2.13.1
- evaluate
- peft
- accelerate
- protobuf
- fsspec==2023.9.2
- bitsandbytes>=0.41.3.post2
- fugashi[unidic_lite]
- sacrebleu[ja]
- black
- flake8