Skip to content

Latest commit

 

History

History
5 lines (4 loc) · 293 Bytes

README.md

File metadata and controls

5 lines (4 loc) · 293 Bytes

Assistant

Fine-tunned GPT2 model to make a personal AI assistant.

pretrained_model.py loads GPT2LMHeadModel, GPT2Tokenizer model from huggingface transformers library and saves it. Then main.py finetune the model taking data from Data folder(Which is not present here because of its size).