winddude commited on
Commit
a02ea5b
1 Parent(s): 9583a07

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -1
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
+ ```