blather 1.0.0

Train and use generative language models with only three lines of code.

Train and use generative text models in a few lines of code.

Installation

Use the package manager pip to install blather.

pip install blather

Fine tuning large generative language models in tractable time require the use of at least a single GPU. For the time being running blather on a CPU only system will cause errors. This may be addressed in the future

Usage

from blather import Blather

blather = Blather()

# fine tunes an appropriate model on your dataset
blather.read(['Text Examples One', 'Second Text Example'... 'Text Example k'])

# returns a text sample generated from the model
blather.write('Sample text to complete')

# saves model
blather.save('model.pt')

# load model from previous training
blather.load('model.pt')

Contributing

https://github.com/bigthonk/blather

Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.

Please make sure to update tests as appropriate.

License

Apache