Skip to content

Commit

Permalink
Update Perceiver AR section in inference examples
Browse files Browse the repository at this point in the history
- fixes #35
  • Loading branch information
krasserm committed Feb 21, 2023
1 parent 5823bd1 commit 9d6a82b
Show file tree
Hide file tree
Showing 5 changed files with 8,933 additions and 7,973 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@ See [Docker image](docs/docker-image.md) for details.
- [Model construction](docs/model-construction.md)
- [Pretrained models](docs/pretrained-models.md)
- [Training examples](docs/training-examples.md)
- [Inference examples](examples/inference.ipynb) [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/krasserm/perceiver-io/blob/0.7.0/examples/inference.ipynb)
- [Inference examples](examples/inference.ipynb) [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/krasserm/perceiver-io/blob/0.8.0/examples/inference.ipynb)
- [Building blocks](docs/building-blocks.md)

## Articles
Expand Down
2 changes: 1 addition & 1 deletion docs/getting-started.md
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ torch.save(model.state_dict(), "/path/to/model.pt")

For generating text from a prompt via top-k sampling, `CausalLanguageModel` provides a `generate()` method. The following
example first loads a trained model from a checkpoint and then generates text from a short sample prompt. An interactive
demo is also available in the [Colab notebook](https://colab.research.google.com/github/krasserm/perceiver-io/blob/0.7.0/examples/inference.ipynb).
demo is also available in the [Colab notebook](https://colab.research.google.com/github/krasserm/perceiver-io/blob/0.8.0/examples/inference.ipynb).

```python
from perceiver.data.text import TextPreprocessor
Expand Down
Loading

0 comments on commit 9d6a82b

Please sign in to comment.