diff --git a/pyproject.toml b/pyproject.toml index 47732bf6..589bce46 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,6 +39,7 @@ backoff = "2.2.1" datasets = "*" optimum = "1.15.0" diffusers = "*" +toml = "*" PyPDF2 = "3.0.1" accelerate = "*" anthropic = "*" diff --git a/requirements.txt b/requirements.txt index 5a9d380c..00c92023 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,6 +19,7 @@ accelerate==0.22.0 chromadb==0.4.14 tensorflow optimum +toml tiktoken==0.4.0 colored addict