chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
"""
|
||||
Sequences module tests
|
||||
"""
|
||||
|
||||
import unittest
|
||||
|
||||
from txtai.pipeline import Sequences
|
||||
|
||||
|
||||
class TestSequences(unittest.TestCase):
|
||||
"""
|
||||
Sequences tests.
|
||||
"""
|
||||
|
||||
def testGeneration(self):
|
||||
"""
|
||||
Test text2text pipeline generation
|
||||
"""
|
||||
|
||||
model = Sequences("t5-small")
|
||||
self.assertEqual(model("Testing the model", prefix="translate English to German: "), "Das Modell zu testen")
|
||||
Reference in New Issue
Block a user