Update README.md
Browse files
README.md
CHANGED
@@ -1,14 +1,12 @@
|
|
1 |
# SliceX AI™ ELM (Efficient Language Models)
|
2 |
This repository contains code to run our ELM models.
|
3 |
|
4 |
-
Models are located in the "models" folder. ELM models in this repository comes in three sizes (elm-1.0, elm-0.75 and elm-0.25) and supports the following use-
|
5 |
- news_classification
|
6 |
-
- toxicity_detection
|
7 |
-
- news_content_generation
|
8 |
|
9 |
## Download ELM repo
|
10 |
```bash
|
11 |
-
git clone git@hf.co:slicexai/elm-
|
12 |
sudo apt-get intall git-lfs
|
13 |
git lfs install
|
14 |
```
|
@@ -20,12 +18,12 @@ PATH=$PATH:/<absolute-path>/git-lfs-3.2.0/
|
|
20 |
git lfs install
|
21 |
```
|
22 |
|
23 |
-
## Download ELM
|
24 |
```bash
|
25 |
-
cd elm-
|
|
|
|
|
26 |
git lfs pull -I models/elm-0.25_news_classification/ckpt.pt
|
27 |
-
git lfs pull -I models/elm-0.25_toxicity_detection/ckpt.pt
|
28 |
-
git lfs pull -I models/elm-0.25_news_content_generation/ckpt.pt
|
29 |
```
|
30 |
|
31 |
## Installation
|
@@ -36,9 +34,9 @@ pip install -r requirements.txt
|
|
36 |
## How to use - Run ELM on a sample task (e.g., news classification)
|
37 |
```bash
|
38 |
python run.py <elm-model-directory>
|
39 |
-
E.g. python run.py models/elm-0.
|
40 |
```
|
41 |
-
Prompts for the specific tasks can be found in the corresponding checkpoint directory. See an example below in the form of `models/elm-0.
|
42 |
```json
|
43 |
{
|
44 |
"inputs": ["GM May Close Plant in Europe DETROIT (Reuters) - General Motors Corp. <A HREF=\"http://www.investor.reuters.com/FullQuote.aspx?ticker=GM.N target=/stocks/quickinfo/fullquote\">GM.N</A> will likely cut some jobs in Europe and may close a plant there as part of a restructuring plan under development to try to return the region to profitability, the U.S. automaker said on Wednesday."],
|
|
|
1 |
# SliceX AI™ ELM (Efficient Language Models)
|
2 |
This repository contains code to run our ELM models.
|
3 |
|
4 |
+
Models are located in the "models" folder. ELM models in this repository comes in three sizes (elm-1.0, elm-0.75 and elm-0.25) and supports the following use-case.
|
5 |
- news_classification
|
|
|
|
|
6 |
|
7 |
## Download ELM repo
|
8 |
```bash
|
9 |
+
git clone git@hf.co:slicexai/elm-v0.1_news_classification
|
10 |
sudo apt-get intall git-lfs
|
11 |
git lfs install
|
12 |
```
|
|
|
18 |
git lfs install
|
19 |
```
|
20 |
|
21 |
+
## Download ELM model checkpoints
|
22 |
```bash
|
23 |
+
cd elm-v0.1_news_classification
|
24 |
+
git lfs pull -I models/elm-1.0_news_classification/ckpt.pt
|
25 |
+
git lfs pull -I models/elm-0.75_news_classification/ckpt.pt
|
26 |
git lfs pull -I models/elm-0.25_news_classification/ckpt.pt
|
|
|
|
|
27 |
```
|
28 |
|
29 |
## Installation
|
|
|
34 |
## How to use - Run ELM on a sample task (e.g., news classification)
|
35 |
```bash
|
36 |
python run.py <elm-model-directory>
|
37 |
+
E.g. python run.py models/elm-0.75_news_classification
|
38 |
```
|
39 |
+
Prompts for the specific tasks can be found in the corresponding checkpoint directory. See an example below in the form of `models/elm-0.75_news_classification/example_prompts.json`.
|
40 |
```json
|
41 |
{
|
42 |
"inputs": ["GM May Close Plant in Europe DETROIT (Reuters) - General Motors Corp. <A HREF=\"http://www.investor.reuters.com/FullQuote.aspx?ticker=GM.N target=/stocks/quickinfo/fullquote\">GM.N</A> will likely cut some jobs in Europe and may close a plant there as part of a restructuring plan under development to try to return the region to profitability, the U.S. automaker said on Wednesday."],
|