Commit
·
bdfe289
1
Parent(s):
6417ddc
Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
language: fr
|
3 |
+
license: mit
|
4 |
+
tags:
|
5 |
+
- causal-lm
|
6 |
+
- fr
|
7 |
+
datasets:
|
8 |
+
- fr_covid_news
|
9 |
+
---
|
10 |
+
|
11 |
+
### GPT-J COVID-19 French News with 8-bit weights
|
12 |
+
|
13 |
+
|
14 |
+
This is a version of Cedille's GPT-J (fr-boris) with 6 billion parameters fine-tuned on [COVID-19 French News dataset](https://huggingface.co/datasets/gustavecortal/fr_covid_news) to generate headlines related to COVID-19.
|
15 |
+
|
16 |
+
You can generate the model in colab or equivalent desktop gpu (e.g. single 1080Ti) because the model has 8-bit weights. Inspired by [GPT-J 8bit](https://huggingface.co/hivemind/gpt-j-6B-8bit).
|
17 |
+
|
18 |
+
Here's how to run it: [![colab](https://camo.githubusercontent.com/84f0493939e0c4de4e6dbe113251b4bfb5353e57134ffd9fcab6b8714514d4d1/68747470733a2f2f636f6c61622e72657365617263682e676f6f676c652e636f6d2f6173736574732f636f6c61622d62616467652e737667)](https://colab.research.google.com/drive/1lMja-CPc0vm5_-gXNXAWU-9c0nom7vZ9)
|
19 |
+
|
20 |
+
This model can be easily loaded using the `GPTJForCausalLM` functionality:
|
21 |
+
```python
|
22 |
+
from transformers import GPTJForCausalLM
|
23 |
+
model = GPTJForCausalLM.from_pretrained("gustavecortal/gpt-j-fr-covid-news")
|
24 |
+
```
|
25 |
+
|
26 |
+
Remember, you have to Monkey-Patch the model before loading it (see Colab above).
|
27 |
+
|
28 |
+
## fr-boris
|
29 |
+
|
30 |
+
Boris is a 6B parameter autoregressive language model based on the GPT-J architecture and trained using the [mesh-transformer-jax](https://github.com/kingoflolz/mesh-transformer-jax) codebase.
|
31 |
+
|
32 |
+
Boris was trained on around 78B tokens of French text from the [C4](https://huggingface.co/datasets/c4) dataset.
|
33 |
+
|
34 |
+
## Links
|
35 |
+
|
36 |
+
* [Gustave Cortal](https://twitter.com/gustavecortal)
|