Skip to content
Snippets Groups Projects
  • Alexandru-Mihai GHERGHESCU's avatar
    accab39c
    Add inference code · accab39c
    Alexandru-Mihai GHERGHESCU authored
    Inference example code. At the moment, the code simply loads a model
    state file and generates text using that. Parameters like max sequence
    length, whether training used fp16, what the tokenizer used for training
    is etc., need to be passed manually by the user (there's a lot of room
    for error here). To be improved.
    
    Merges changes from !14
    Closes !14
    accab39c
    History
    Add inference code
    Alexandru-Mihai GHERGHESCU authored
    Inference example code. At the moment, the code simply loads a model
    state file and generates text using that. Parameters like max sequence
    length, whether training used fp16, what the tokenizer used for training
    is etc., need to be passed manually by the user (there's a lot of room
    for error here). To be improved.
    
    Merges changes from !14
    Closes !14
Code owners