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 |
+
```
|