From 3e692ed2544584054b097e53f5202eac18fcf688 Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 31 Jan 2024 21:19:59 -0500 Subject: [PATCH] [REQUIREMENT][tool] --- pyproject.toml | 1 + requirements.txt | 1 + 2 files changed, 2 insertions(+) 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