hungnm commited on
Commit
8a91a1c
·
verified ·
1 Parent(s): fddfffd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +45 -44
README.md CHANGED
@@ -1,44 +1,44 @@
1
- ---
2
- library_name: transformers
3
- license: mit
4
- base_model: jhu-clsp/mmBERT-small
5
- tags:
6
- - sentiment
7
- - text-classification
8
- - multilingual
9
- - modernbert
10
- - sentiment-analysis
11
- - product-reviews
12
- - place-reviews
13
- - mmbert
14
- metrics:
15
- - f1
16
- - precision
17
- - recall
18
- model-index:
19
- - name: mmBERT-small-multilingual-sentiment
20
- results: []
21
- datasets:
22
- - clapAI/MultiLingualSentiment
23
- language:
24
- - en
25
- - zh
26
- - vi
27
- - ko
28
- - ja
29
- - ar
30
- - de
31
- - es
32
- - fr
33
- - hi
34
- - id
35
- - it
36
- - ms
37
- - pt
38
- - ru
39
- - tr
40
- pipeline_tag: text-classification
41
- ---
42
 
43
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
44
  should probably proofread and complete it, then remove this comment. -->
@@ -87,13 +87,14 @@ pip install transformers==4.55.0
87
  `Optional: accelerate inference with FlashAttention-2 (if supported by your GPU):`
88
 
89
  ```bash
90
- pip install packaging
91
- pip install ninja
92
- MAX_JOBS=4 pip install flash-attn --no-build-isolation
93
  ```
94
 
95
  ### Example Usage
96
 
 
 
97
  ```python
98
  import torch
99
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
1
+ ---
2
+ library_name: transformers
3
+ license: mit
4
+ base_model: jhu-clsp/mmBERT-small
5
+ tags:
6
+ - sentiment
7
+ - text-classification
8
+ - multilingual
9
+ - modernbert
10
+ - sentiment-analysis
11
+ - product-reviews
12
+ - place-reviews
13
+ - mmbert
14
+ metrics:
15
+ - f1
16
+ - precision
17
+ - recall
18
+ model-index:
19
+ - name: mmBERT-small-multilingual-sentiment
20
+ results: []
21
+ datasets:
22
+ - clapAI/MultiLingualSentiment
23
+ language:
24
+ - en
25
+ - zh
26
+ - vi
27
+ - ko
28
+ - ja
29
+ - ar
30
+ - de
31
+ - es
32
+ - fr
33
+ - hi
34
+ - id
35
+ - it
36
+ - ms
37
+ - pt
38
+ - ru
39
+ - tr
40
+ pipeline_tag: text-classification
41
+ ---
42
 
43
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
44
  should probably proofread and complete it, then remove this comment. -->
 
87
  `Optional: accelerate inference with FlashAttention-2 (if supported by your GPU):`
88
 
89
  ```bash
90
+ pip install packaging==25.0 ninja==1.13.0
91
+ MAX_JOBS=4 pip install flash-attn==2.8.3 --no-build-isolation
 
92
  ```
93
 
94
  ### Example Usage
95
 
96
+ Try it on [Google Colab](https://colab.research.google.com/drive/1nsh22sEz0znV3OedE8RqA0dNoPOJjghJ?usp=sharing)
97
+
98
  ```python
99
  import torch
100
  from transformers import AutoTokenizer, AutoModelForSequenceClassification