How to use a fine-tuned Checkpoint?

salesforce / CodeT5

Home of CodeT5: Open Code LLMs for Code Understanding and Generation

https://arxiv.org/abs/2305.07922

BSD 3-Clause "New" or "Revised" License

2.68k stars 394 forks source link

How to use a fine-tuned Checkpoint? #34

Closed KareemAlaa2001 closed 2 years ago

KareemAlaa2001 commented 2 years ago

Hi,

This might seem like a basic question, but I want to be sure I'm on the right track here. I am trying to use the code generation fine-tuned checkpoint, but since the binary file isn't sufficient to load the model in huggingface, I downloaded the huggingface codet5-base directory and replaced the pythorch_model.bin with the binary file corresponding to the concode task. Is this the correct way to go about this?

yuewang-cuhk commented 2 years ago

Yes, this is one correct way. Or you can follow another way here

KareemAlaa2001 commented 2 years ago

Cool, thanks for weighing in!