FrostAura's picture
[DEV]: Adding of logo to README file.
2201472
|
raw
history blame
1.83 kB
metadata
language:
  - en
thumbnail: >-
  https://github.com/faGH/fa.creative/blob/master/Icons/FrostAura/FA%20Logo/FrostAura.Logo.Complex.png?raw=true
tags:
  - text-generation
  - novel-generation
  - fiction
  - gpt-neo-x
  - pytorch
license: mit

fa.intelligence.models.generative.novels.fiction

Description

This FrostAura Intelligence is a fine-tuned version of EleutherAI/gpt-neox-20b for fictional text content generation.

Getting Started

PIP Installation

pip install -U --no-cache-dir transformers

Usage

from transformers import GPTNeoXForCausalLM, GPTNeoXTokenizerFast

model_name = 'FrostAura/gpt-neox-20b-fiction-novel-generation'
model = GPTNeoXForCausalLM.from_pretrained(model_name)
tokenizer = GPTNeoXTokenizerFast.from_pretrained(model_name)

prompt = 'GPTNeoX20B is a 20B-parameter autoregressive Transformer model developed by EleutherAI.'

input_ids = tokenizer(prompt, return_tensors="pt").input_ids

gen_tokens = model.generate(
  input_ids,
  do_sample=True,
  temperature=0.9,
  max_length=100,
)
gen_text = tokenizer.batch_decode(gen_tokens)[0]

print(f'Result: {gen_text}')

Contribute

In order to contribute, simply fork the repository, make changes and create a pull request.

Support

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.

Supported Platforms Link
PayPal Donate via Paypal

For any queries, contact dean.martin@frostaura.net.