RAG-FiT/pyproject.toml at main · IntelLabs/RAG-FiT · GitHub

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
[project]
name = "ragfit"
version = "1.2.0"
description = "Framework for enhancing LLMs for RAG tasks using fine-tuning."
readme = "README.md"
license = {file = "LICENSE"}
requires-python = ">=3.10"
dependencies = [
    "bert-score>=0.3.13",
    "bitsandbytes==0.42.0",
    "datasets==2.16.1",
    "evaluate==0.4.1",
    "hydra-core==1.3.2",
    "nltk==3.9",
    "openai==1.23.3",
    "peft==0.11.1",
    "pyyaml==6.0.1",
    "rouge-score==0.1.2",
    "sentence-transformers==2.4.0",
    "sentencepiece==0.2.0",
    "torch>=2.8.0",
    "transformers>=4.50.0",
    "trl==0.8.6",
    "wandb==0.16.4",
]

[project.urls]
Homepage = "https://github.com/IntelLabs/RAG-FiT"
Documentation = "https://intellabs.github.io/RAG-FiT/"

[tool.setuptools]
packages = ["ragfit"]

[project.optional-dependencies]
deepeval = [
    "deepeval==0.21.73",
]
haystack = [
    "haystack-ai==2.3.1",
    "qdrant-haystack>=5.0.0",
]