EnergonAI icon indicating copy to clipboard operation
EnergonAI copied to clipboard

OPT inference generate example

Open virgulvirgul opened this issue 2 years ago • 1 comments

Hi, is there any generate example for OTP models?

virgulvirgul avatar Aug 06 '22 13:08 virgulvirgul

Hi, you can follow https://github.com/dujiangsu/EnergonAI/blob/feature/opt-175b/energonai/utils/checkpointing_opt.py to load parameters of opt model. As for the generate example, you can refer to the hf_gpt2 example.

dujiangsu avatar Aug 08 '22 06:08 dujiangsu