Spaces:
Runtime error
Runtime error
Update app_dialogue.py
Browse files- 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 |
)
|