Update README.md
Browse filesAdding of README file.
README.md
CHANGED
@@ -1,3 +1,58 @@
|
|
1 |
---
|
2 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
+
language:
|
3 |
+
- en
|
4 |
+
thumbnail: "https://github.com/faGH/fa.creative/blob/master/Icons/FrostAura/FA%20Logo/FrostAura.Logo.Complex.png?raw=true"
|
5 |
+
tags:
|
6 |
+
- text-generation
|
7 |
+
- machine-learning
|
8 |
+
- deep-learning
|
9 |
+
- pytorch
|
10 |
+
license: "mit"
|
11 |
+
datasets:
|
12 |
+
- na
|
13 |
+
metrics:
|
14 |
+
- na
|
15 |
---
|
16 |
+
# fa.intelligence.models.generative.novels.fiction
|
17 |
+
## Description
|
18 |
+
This FrostAura Intelligence is a fine-tuned version of [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b) for fictional text content generation.
|
19 |
+
|
20 |
+
## Getting Started
|
21 |
+
### PIP Installation
|
22 |
+
```
|
23 |
+
pip install -U --no-cache-dir transformers
|
24 |
+
```
|
25 |
+
### Usage
|
26 |
+
```
|
27 |
+
from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast
|
28 |
+
|
29 |
+
model_name = 'FrostAura/gpt-neox-20b-fiction-novel-generation'
|
30 |
+
model = GPTNeoXForCausalLM.from_pretrained(model_name)
|
31 |
+
tokenizer = GPTNeoXTokenizerFast.from_pretrained(model_name)
|
32 |
+
|
33 |
+
prompt = 'GPTNeoX20B is a 20B-parameter autoregressive Transformer model developed by EleutherAI.'
|
34 |
+
|
35 |
+
input_ids = tokenizer(prompt, return_tensors="pt").input_ids
|
36 |
+
|
37 |
+
gen_tokens = model.generate(
|
38 |
+
input_ids,
|
39 |
+
do_sample=True,
|
40 |
+
temperature=0.9,
|
41 |
+
max_length=100,
|
42 |
+
)
|
43 |
+
gen_text = tokenizer.batch_decode(gen_tokens)[0]
|
44 |
+
|
45 |
+
print(f'Result: {gen_text}')
|
46 |
+
```
|
47 |
+
|
48 |
+
## Contribute
|
49 |
+
In order to contribute, simply fork the repository, make changes and create a pull request.
|
50 |
+
|
51 |
+
## Support
|
52 |
+
If you enjoy FrostAura open-source content and would like to support us in continuous delivery, please consider a donation via a platform of your choice.
|
53 |
+
|
54 |
+
| Supported Platforms | Link |
|
55 |
+
| ------------------- | ---- |
|
56 |
+
| PayPal | [Donate via Paypal](https://www.paypal.com/donate/?hosted_button_id=SVEXJC9HFBJ72) |
|
57 |
+
|
58 |
+
For any queries, contact dean.martin@frostaura.net.
|