From e29d6511f5b6fadee5d0779fe477d039acef8003 Mon Sep 17 00:00:00 2001 From: younesbelkada Date: Tue, 4 Apr 2023 07:06:57 +0000 Subject: [PATCH] more description --- tests/test_common_gpu.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_common_gpu.py b/tests/test_common_gpu.py index d9099ce..cf1fad9 100644 --- a/tests/test_common_gpu.py +++ b/tests/test_common_gpu.py @@ -31,7 +31,9 @@ if is_bnb_available(): @require_torch_gpu class PeftGPUCommonTests(unittest.TestCase): - r""" """ + r""" + A common tester to run common operations that are performed on GPU such as generation, loading in 8bit, etc. + """ def setUp(self): self.seq2seq_model_id = "google/flan-t5-base"