gustavecortal commited on
Commit
bdfe289
·
1 Parent(s): 6417ddc

Create README.md

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