Ali-C137 commited on
Commit
24e8c3d
1 Parent(s): a7cbbd8

Update app_dialogue.py

Browse files
Files changed (1) hide show
  1. app_dialogue.py +10 -0
app_dialogue.py CHANGED
@@ -1,9 +1,19 @@
1
  import os
2
  import subprocess
 
 
 
 
 
 
 
 
 
3
 
4
  # Install flash attention
5
  subprocess.run(
6
  "pip install flash-attn --no-build-isolation",
 
7
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
8
  shell=True,
9
  )
 
1
  import os
2
  import subprocess
3
+ import datasets
4
+ from langchain.docstore.document import Document
5
+ from langchain.text_splitter import RecursiveCharacterTextSplitter
6
+ from langchain.vectorstores import FAISS
7
+ from langchain_community.embeddings import HuggingFaceEmbeddings
8
+ import json
9
+ from transformers.agents import Tool
10
+ from langchain_core.vectorstores import VectorStore
11
+ from transformers.agents import HfEngine, ReactJsonAgent
12
 
13
  # Install flash attention
14
  subprocess.run(
15
  "pip install flash-attn --no-build-isolation",
16
+ "pip install git+https://github.com/huggingface/transformers.git#egg=transformers[agents]",
17
  env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
18
  shell=True,
19
  )