Mixtral import

artitw · Feb 10, 2024 · 84f4b31 · 84f4b31
1 parent dac86f5
commit 84f4b31
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.4.0",
+  version="1.4.1",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",

diff --git a/text2text/assistant.py b/text2text/assistant.py
@@ -116,4 +116,6 @@ def chat_completion(self, messages=[{"role": "user", "content": "hello"}], strea
     return results
 
   def transform(self, input_lines, src_lang='en', **kwargs):
-    return self.chat_completion([{"role": "user", "content": input_lines}])
+    return self.chat_completion([{"role": "user", "content": input_lines}])
+
+  completion = transform
diff --git a/text2text/mixtral/__init__.py b/text2text/mixtral/__init__.py
@@ -0,0 +1 @@
+from .build_model import OffloadConfig, QuantConfig, build_model