Update README.md
Browse files
README.md
CHANGED
@@ -4,4 +4,28 @@ datasets:
|
|
4 |
- winddude/stock_price_chat_ds
|
5 |
language:
|
6 |
- en
|
7 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
- winddude/stock_price_chat_ds
|
5 |
language:
|
6 |
- en
|
7 |
+
---
|
8 |
+
|
9 |
+
# Stock Price Chat Lora
|
10 |
+
|
11 |
+
[GitHub](https://github.com/getorca/stock_price_chat) | [Blog](https://nootka.ai)
|
12 |
+
|
13 |
+
Stock Price Chat is an intent/action model. It is an experiment of a type of RaG(Retrevial augmented generation) for answer plain text querries for stock prices.
|
14 |
+
|
15 |
+
### Usage
|
16 |
+
|
17 |
+
1) download the base Llama2 7b model
|
18 |
+
2) replace the tokenizer with the tokenizers in this repo
|
19 |
+
3) when loading a model, after loading the tokenizer make sure to call `model.resize_token_embeddings(len(stokenizer))`
|
20 |
+
4) with peft to load the adapter in this repo.
|
21 |
+
|
22 |
+
The model needs to be augmented with knowledge for yFinance, so use code found here: <https://github.com/getorca/stock_price_chat>. More details on the archetecture of the intent/action loop are also available here.
|
23 |
+
|
24 |
+
### Basic Prompt Format
|
25 |
+
```
|
26 |
+
<|SYSTEM|>You are a bot that provides stock prices. From a user input first create an action with the ticker and date in a jsons string. If you are sent an action and knowledge create the response with the stock price from the provided knowledge for the date the user asks.<|END_SYSTEM|>
|
27 |
+
<|INPUT|>user input<|END_INPUT|>
|
28 |
+
<|ACTION|>action string generated by the model<|END_ACTION|>
|
29 |
+
<|KNOWLEDGE|>knowledge string returned via the api call<|END_KNOWLEDGE|>
|
30 |
+
<|RESPONSE|>plain text response generated by the model<|END_RESPONSE|>
|
31 |
+
```
|