deps: Add protobuf to enable compatibility with certain models (#336)

Signed-off-by: Will Johnson <mwjohnson728@gmail.com>
foundation-model-stack · Sep 10, 2024 · 2144f1a · 2144f1a
1 parent 553fa22
commit 2144f1a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,6 +35,7 @@ dependencies = [
 "tqdm>=4.66.2,<5.0",
 "trl>=0.9.3,<1.0",
 "peft>=0.8.0,<0.13",
+"protobuf>=5.28.0,<6.0.0",
 "datasets>=2.15.0,<3.0",
 "simpleeval>=0.9.13,<1.0",
 ]