Clean up dependencies and demo

artitw · Jul 24, 2024 · 2960b19 · 2960b19
1 parent 2dc8e5c
commit 2960b19
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/demos/Text2Text_LLM.ipynb b/demos/Text2Text_LLM.ipynb
@@ -111,7 +111,7 @@
         "  print()\n",
         "  messages.append({\"role\": \"user\", \"content\": user_input})\n",
         "  print(\"Assistant: \")\n",
-        "  result = asst.chat_completion(messages, stream=False, max_new_tokens=8)\n",
+        "  result = asst.chat_completion(messages, stream=False)\n",
         "  print(result[\"message\"][\"content\"])\n",
         "  messages.append(result[\"message\"])\n",
         "  print()"

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.4.5",
+  version="1.4.6",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",
@@ -20,24 +20,19 @@
   ],
   keywords='multilingual crosslingual gpt chatgpt bert natural language processing nlp nlg text generation gpt question answer answering information retrieval tfidf tf-idf bm25 search index summary summarizer summarization tokenizer tokenization translation backtranslation data augmentation science machine learning colab embedding levenshtein sub-word edit distance conversational dialog chatbot mixtral',
   install_requires=[
-    'accelerate',
-    'bitsandbytes',
-    'peft',
     'faiss-cpu',
     'flask',
-    'langchain',
     'googledrivedownloader',
     'numpy',
     'ollama',
-    'optimum',
     'pandas',
     'scikit-learn',
     'scipy',
     'sentencepiece',
     'torch',
     'tqdm',
     'transformers',
+    'huggingface_hub',
     'hqq',
-    'huggingface_hub'
   ],
 )