try-gptj-generation

A wrapper to simply load GPT-J and use it for generation. Uses DeepSpeed ons stage 2 or 3 for inference, …

Stars

6

Forks

3

Language

Python

Last Updated

May 01, 2023

Similar Repos