Skip to content
Snippets Groups Projects
Unverified Commit 209826e4 authored by Alexandru-Mihai GHERGHESCU's avatar Alexandru-Mihai GHERGHESCU
Browse files

README cleanup

parent a91b0d2a
No related branches found
No related tags found
1 merge request!25Re-factor optimus-prime code (optimus-prime v2)
# Optimus Prime # Optimus Prime
(Yet another) PyTorch framework for training large language models. (Yet another) PyTorch framework for training large language models
## How to use ## How to use
...@@ -17,7 +17,7 @@ with adjustable top-p threshold and temperature values. ...@@ -17,7 +17,7 @@ with adjustable top-p threshold and temperature values.
## Basic building blocks ## Basic building blocks
As its parent PyTorch, the framework is split between a number of modules. The As PyTorch itself, this framework is split between a number of modules. The
most important modules are the `OptimusDataLoader`, the `Dataset`s, the most important modules are the `OptimusDataLoader`, the `Dataset`s, the
`Trainer`, the tokenizers and the models. These can be combined and adapted in `Trainer`, the tokenizers and the models. These can be combined and adapted in
any way, shape or form to train a model from scratch. any way, shape or form to train a model from scratch.
...@@ -45,7 +45,7 @@ Of course, any number of the above can be used as defaults. ...@@ -45,7 +45,7 @@ Of course, any number of the above can be used as defaults.
There are a number of packages required to run the framework. Get your closest There are a number of packages required to run the framework. Get your closest
Python retailer and ask him to run the following command: Python retailer and ask him to run the following command:
`pip install torch fire sentencepiece fastprogress matplotlib` `pip install torch fire sentencepiece fastprogress`
## License ## License
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment