Commit
·
24ce340
1
Parent(s):
effad5d
[add]: about us / more information model card description
Browse files
README.md
CHANGED
@@ -5,9 +5,7 @@ base_model: tiiuae/falcon-7b-instruct
|
|
5 |
|
6 |
# Model Card for the Query Parser LLM using Falcon-7B-Instruct
|
7 |
|
8 |
-
[]()
|
9 |
-
[](https://www.python.org/downloads/release/python-390/)
|
10 |
-

|
11 |
|
12 |
EmbeddingStudio is the [open-source framework](https://github.com/EulerSearch/embedding_studio/tree/main), that allows you transform a joint "Embedding Model + Vector DB" into
|
13 |
a full-cycle search engine: collect clickstream -> improve search experience-> adapt embedding model and repeat out of the box.
|
@@ -45,9 +43,9 @@ This is only [Falcon-7B-Instruct](https://huggingface.co/tiiuae/falcon-7b-instru
|
|
45 |
**Important:** Additionally, we are trying to fine-tune the Large Language Model (LLM) to not only parse unstructured search queries but also to correct spelling.
|
46 |
|
47 |
- **Developed by EmbeddingStudio team:**
|
48 |
-
* Aleksandr Iudaev
|
49 |
-
* Andrei Kostin
|
50 |
-
* ML Doom
|
51 |
- **Funded by EmbeddingStudio team**
|
52 |
- **Model type:** Instruct Fine-Tuned Large Language Model
|
53 |
- **Model task:** Zero-shot search query parsing
|
@@ -169,7 +167,6 @@ INSTRUCTION_TEMPLATE = """
|
|
169 |
"""
|
170 |
|
171 |
|
172 |
-
|
173 |
def parse(
|
174 |
query: str,
|
175 |
company_category: str,
|
@@ -201,15 +198,15 @@ def parse(
|
|
201 |
|
202 |
## Bias, Risks, and Limitations
|
203 |
|
204 |
-
|
|
|
205 |
|
206 |
-
|
|
|
207 |
|
208 |
### Recommendations
|
209 |
|
210 |
-
<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
|
211 |
|
212 |
-
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
|
213 |
|
214 |
## How to Get Started with the Model
|
215 |
|
@@ -310,23 +307,51 @@ Carbon emissions can be estimated using the [Machine Learning Impact calculator]
|
|
310 |
|
311 |
#### Software
|
312 |
|
313 |
-
|
|
|
|
|
|
|
314 |
|
315 |
-
|
316 |
|
317 |
-
|
|
|
|
|
|
|
318 |
|
319 |
-
[
|
320 |
|
321 |
-
|
322 |
|
323 |
-
|
|
|
|
|
|
|
|
|
|
|
324 |
|
325 |
-
|
326 |
|
327 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
328 |
|
|
|
|
|
|
|
329 |
|
330 |
### Framework versions
|
331 |
|
332 |
-
- PEFT 0.
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
# Model Card for the Query Parser LLM using Falcon-7B-Instruct
|
7 |
|
8 |
+
[]() [](https://www.python.org/downloads/release/python-390/) 
|
|
|
|
|
9 |
|
10 |
EmbeddingStudio is the [open-source framework](https://github.com/EulerSearch/embedding_studio/tree/main), that allows you transform a joint "Embedding Model + Vector DB" into
|
11 |
a full-cycle search engine: collect clickstream -> improve search experience-> adapt embedding model and repeat out of the box.
|
|
|
43 |
**Important:** Additionally, we are trying to fine-tune the Large Language Model (LLM) to not only parse unstructured search queries but also to correct spelling.
|
44 |
|
45 |
- **Developed by EmbeddingStudio team:**
|
46 |
+
* Aleksandr Iudaev [[LinkedIn](https://www.linkedin.com/in/alexanderyudaev/)] [[Email](mailto:[email protected])]
|
47 |
+
* Andrei Kostin [[LinkedIn](https://www.linkedin.com/in/andrey-kostin/)] [[Email](mailto:[email protected])]
|
48 |
+
* ML Doom [AI Assistant]
|
49 |
- **Funded by EmbeddingStudio team**
|
50 |
- **Model type:** Instruct Fine-Tuned Large Language Model
|
51 |
- **Model task:** Zero-shot search query parsing
|
|
|
167 |
"""
|
168 |
|
169 |
|
|
|
170 |
def parse(
|
171 |
query: str,
|
172 |
company_category: str,
|
|
|
198 |
|
199 |
## Bias, Risks, and Limitations
|
200 |
|
201 |
+
### Bias
|
202 |
+
|
203 |
|
204 |
+
|
205 |
+
### Risks
|
206 |
|
207 |
### Recommendations
|
208 |
|
|
|
209 |
|
|
|
210 |
|
211 |
## How to Get Started with the Model
|
212 |
|
|
|
307 |
|
308 |
#### Software
|
309 |
|
310 |
+
* Python 3.9+
|
311 |
+
* CUDA 11.7.1
|
312 |
+
* NVIDIA [Compatible Drivers](https://www.nvidia.com/download/find.aspx)
|
313 |
+
* Torch 2.0.0
|
314 |
|
315 |
+
## More Information / About us
|
316 |
|
317 |
+
EmbeddingStudio is an innovative open-source framework designed to seamlessly convert a combined
|
318 |
+
"Embedding Model + Vector DB" into a comprehensive search engine. With built-in functionalities for
|
319 |
+
clickstream collection, continuous improvement of search experiences, and automatic adaptation of
|
320 |
+
the embedding model, it offers an out-of-the-box solution for a full-cycle search engine.
|
321 |
|
322 |
+

|
323 |
|
324 |
+
### Features
|
325 |
|
326 |
+
1. 🔄 Turn your vector database into a full-cycle search engine
|
327 |
+
2. 🖱️ Collect users feedback like clickstream
|
328 |
+
3. 🚀 (*) Improve search experience on-the-fly without frustrating wait times
|
329 |
+
4. 📊 (*) Monitor your search quality
|
330 |
+
5. 🎯 Improve your embedding model through an iterative metric fine-tuning procedure
|
331 |
+
6. 🆕 (*) Use the new version of the embedding model for inference
|
332 |
|
333 |
+
(*) - features in development
|
334 |
|
335 |
+
EmbeddingStudio is highly customizable, so you can bring your own:
|
336 |
+
|
337 |
+
1. Data source
|
338 |
+
2. Vector database
|
339 |
+
3. Clickstream database
|
340 |
+
4. Embedding model
|
341 |
+
|
342 |
+
For more details visit [GitHub Repo](https://github.com/EulerSearch/embedding_studio/tree/main).
|
343 |
+
|
344 |
+
## Model Card Authors and Contact
|
345 |
|
346 |
+
* Aleksandr Iudaev [[LinkedIn](https://www.linkedin.com/in/alexanderyudaev/)] [[Email](mailto:[email protected])]
|
347 |
+
* Andrei Kostin [[LinkedIn](https://www.linkedin.com/in/andrey-kostin/)] [[Email](mailto:[email protected])]
|
348 |
+
* ML Doom [AI Assistant]
|
349 |
|
350 |
### Framework versions
|
351 |
|
352 |
+
- PEFT 0.5.0
|
353 |
+
- Datasets 2.16.1
|
354 |
+
- BitsAndBytes 0.41.0
|
355 |
+
- PyTorch 2.0.0
|
356 |
+
- Transformers 4.36.2
|
357 |
+
- TRL 0.7.7
|