more description

This commit is contained in:
younesbelkada
2023-04-04 07:06:57 +00:00
parent 4d3b4ab206
commit e29d6511f5
+3 -1
View File
@@ -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"