doberst commited on
Commit
46cb2e1
1 Parent(s): 76c6d4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -8
README.md CHANGED
@@ -1,19 +1,19 @@
1
  ---
2
- license: cc-by-sa-4.0
3
  inference: false
4
  ---
5
 
6
- # SLIM-EXTRACT
7
 
8
  <!-- Provide a quick summary of what the model is/does. -->
9
 
10
- **slim-extract** implements a specialized function-calling customizable 'extract' capability that takes as an input a context passage, a customized key, and outputs a python dictionary with key that corresponds to the customized key, with a value consisting of a list of items extracted from the text corresponding to that key, e.g.,
11
 
12
  &nbsp;&nbsp;&nbsp;&nbsp;`{'universities': ['Berkeley, Stanford, Yale, University of Florida, ...'] }`
13
 
14
- This model is fine-tuned on top of [**llmware/bling-stable-lm-3b-4e1t-v0**](https://huggingface.co/llmware/bling-stable-lm-3b-4e1t-v0), which in turn, is a fine-tune of stabilityai/stablelm-3b-4elt.
15
 
16
- For fast inference use, we would recommend the 'quantized tool' version, e.g., [**'slim-extract-tool'**](https://huggingface.co/llmware/slim-extract-tool).
17
 
18
 
19
  ## Prompt format:
@@ -27,8 +27,8 @@ For fast inference use, we would recommend the 'quantized tool' version, e.g.,
27
  <details>
28
  <summary>Transformers Script </summary>
29
 
30
- model = AutoModelForCausalLM.from_pretrained("llmware/slim-extract")
31
- tokenizer = AutoTokenizer.from_pretrained("llmware/slim-extract")
32
 
33
  function = "extract"
34
  params = "company"
@@ -70,7 +70,7 @@ For fast inference use, we would recommend the 'quantized tool' version, e.g.,
70
  <summary>Using as Function Call in LLMWare</summary>
71
 
72
  from llmware.models import ModelCatalog
73
- slim_model = ModelCatalog().load_model("llmware/slim-extract")
74
  response = slim_model.function_call(text,params=["company"], function="extract")
75
 
76
  print("llmware - llm_response: ", response)
 
1
  ---
2
+ license: apache-2.0
3
  inference: false
4
  ---
5
 
6
+ # SLIM-EXTRACT-TINY
7
 
8
  <!-- Provide a quick summary of what the model is/does. -->
9
 
10
+ **slim-extract-tiny** implements a specialized function-calling customizable 'extract' capability that takes as an input a context passage, a customized key, and outputs a python dictionary with key that corresponds to the customized key, with a value consisting of a list of items extracted from the text corresponding to that key, e.g.,
11
 
12
  &nbsp;&nbsp;&nbsp;&nbsp;`{'universities': ['Berkeley, Stanford, Yale, University of Florida, ...'] }`
13
 
14
+ This model is fine-tuned on top of a tiny-llama 1b base.
15
 
16
+ For fast inference use, we would recommend the 'quantized tool' version, e.g., [**'slim-extract-tiny-tool'**](https://huggingface.co/llmware/slim-extract-tiny-tool).
17
 
18
 
19
  ## Prompt format:
 
27
  <details>
28
  <summary>Transformers Script </summary>
29
 
30
+ model = AutoModelForCausalLM.from_pretrained("llmware/slim-extract-tiny")
31
+ tokenizer = AutoTokenizer.from_pretrained("llmware/slim-extract-tiny")
32
 
33
  function = "extract"
34
  params = "company"
 
70
  <summary>Using as Function Call in LLMWare</summary>
71
 
72
  from llmware.models import ModelCatalog
73
+ slim_model = ModelCatalog().load_model("llmware/slim-extract-tiny")
74
  response = slim_model.function_call(text,params=["company"], function="extract")
75
 
76
  print("llmware - llm_response: ", response)