From 8b980e852dedecca2d6bf1cf3bde00a2c1168150 Mon Sep 17 00:00:00 2001
From: HelloWorldLTY <43333475+HelloWorldLTY@users.noreply.github.com>
Date: Wed, 26 Jun 2024 21:45:01 -0700
Subject: [PATCH] Update requirements.txt

Add more packages for finetuning
---
 requirements.txt | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index dce85f9..4004b9b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -31,3 +31,6 @@ accelerate==0.30.1
 socksio==1.0.0
 gradio
 gradio_client
+deepspeed==0.14.4
+peft==0.11.1
+tensorboard==2.17.0