Live data from Hacker News

Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

github.com

11–20 of 22 posts

Re: Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

#11

I have been meaning to explore fine-tuning llms on my own dataset. Which formats does this toolkit support?

You can fine-tune on your own dataset! As long as your dataset is in one of json, csv or huggingface formats, our toolkit can ingest your data!

Re: Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

#12

This is a great project, little bit similar to https://github.com/ludwig-ai/ludwig , but it includes testing capabilities and ablation. questions regarding the LLM testing aspect: How extensive is the test coverage for LLM use cases, and what is the current state of this project area? Do you offer any guarantees, or is it considered an open-ended problem? Would love to see more progress toward this direction!

Thanks for the feedback! Yes, it is similar to ludwig but we do think that our toolkit is a more lightweight solution to fine-tuning and ablation studies. In most cases, finding the right LLM with the right config on your dataset requires multiple runs (grid search). Our toolkit offers this capability via one yaml file.

As for the test coverage, right now, the toolkit includes property-based unit tests. For instance, for an LLM fine-tuned on summarization, a property-test will evaluate if the summarized text is smaller in length compared to the actual input text.

Similar to the above test, we have a handful of property-based tests. Of course, the list is not exhaustive at this time. As more progress is being made on the testing side, we aim to distill the most relevant tests depending on use-cases.

Hope this helps.

Re: Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

#14
Sounds like a great library to use for automatically testing all the new models being released everyday and finding out if a new open source model significantly performs better on your custom dataset.

1. What's the largest model (number of parameters) that you've tested the library with?

2. Will MoE models work as well? They're known to have more unstable training and need some custom techniques to stabilize

Re: Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

#17

This is a great project! But I was wondering how frequently will the library be updated with the new different optimization techniques which keep coming out?

Thanks for the feedback! The goal is to offer new techniques via our toolkit as soon as they become available on HuggingFace. To that end, we are aiming to move fast and bring those techniques to the toolkit at the earliest post release.

Re: Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

#18

Sounds like a great library to use for automatically testing all the new models being released everyday and finding out if a new open source model significantly performs better on your custom dataset. 1. What's the largest model (number of parameters) that you've tested the library with? 2. Will MoE models work as well? They're known to have more unstable training and need some custom techniques to stabilize

Thanks for the feedback!

1. The largest model that we have tested is Llama2 13B. For the first phase, we focussed on fine-tuning LLMs in the 1B-13B range. For our next phase, we will focus on 13B-45B'ish -- for this we will have to incorporate distributed techniques.

2. Following incorporation of distributed training techniques, we will be able to run MoE based models, such as Mixtral.

Re: Show HN: Toolkit for LLM Fine-Tuning, Ablating and Testing

#20

Just tried this out and got the default working in a few minutes! Would love to see more support to use finetuning dataset format used for OpenAI and handling for history.

Thanks for the feedback! Glad you got the default setting working quickly!

Right now, we are focussed mostly on offering support for open-source models but we can definitely extend support for OpenAI formats.

May I ask what history means?

Post reply on HN