From 7fb3838c002baa737387b22dc40e50965186be13 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Thu, 18 Jul 2024 10:59:09 -0700
Subject: [PATCH] Update pyproject.toml

---
 pyproject.toml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 7d018b8b79..29b35577e9 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -46,6 +46,7 @@ huggingface = [
     "trl>=0.7.9,<0.9.0",
     "peft>=0.7.1,!=0.11.0",
     "protobuf<4.0.0",
+    "huggingface_hub[hf_transfer]",
 ]
 cu118only = [
     "xformers @ https://download.pytorch.org/whl/cu118/xformers-0.0.22.post7%2Bcu118-cp39-cp39-manylinux2014_x86_64.whl ; python_version=='3.9'",
@@ -195,6 +196,7 @@ colab-new = [
     "wheel>=0.42.0",
     "numpy",
     "protobuf<4.0.0",
+    "huggingface_hub[hf_transfer]",
 ]
 colab-no-deps = [
     "accelerate>=0.26.1",