Fine-tuning Large Language Models (LLMs) | w/ Example Code
👉 Need help with AI? Reach out: shawhintalebi.com/
This is the 5th video in a series on using large language models (LLMs) in practice. Here, I discuss how to fine-tune an existing LLM for a particular use case and walk through a concrete example with Python code.
More Resources:
👉 Series Playlist: • Large Language Models ...
📰 Read more: towardsdatascience.com/fine-t...
💻 Example code: github.com/ShawhinT/KZhead-B...
Final Model: huggingface.co/shawhin/distil...
🔢 Dataset: huggingface.co/datasets/shawh...
[1] Deeplearning.ai Finetuning Large Langauge Models Short Course: www.deeplearning.ai/short-cou...
[2] arXiv:2005.14165 [cs.CL] (GPT-3 Paper)
[3] arXiv:2303.18223 [cs.CL] (Survey of LLMs)
[4] arXiv:2203.02155 [cs.CL] (InstructGPT paper)
[5] 🤗 PEFT: Parameter-Efficient Fine-Tuning of Billion-Scale Models on Low-Resource Hardware: huggingface.co/blog/peft
[6] arXiv:2106.09685 [cs.CL] (LoRA paper)
[7] Original dataset source - Andrew L. Maas, Raymond E. Daly, Peter T. Pham, Dan Huang, Andrew Y. Ng, and Christopher Potts. 2011. Learning Word Vectors for Sentiment Analysis. In Proceedings of the 49th Annual Meeting of the Association for Computational Linguistics: Human Language Technologies, pages 142-150, Portland, Oregon, USA. Association for Computational Linguistics.
--
Book a call: calendly.com/shawhintalebi
Socials
/ shawhin
/ shawhintalebi
/ shawhint
/ shawhintalebi
The Data Entrepreneurs
🎥 KZhead: / @thedataentrepreneurs
👉 Discord: / discord
📰 Medium: / the-data
📅 Events: lu.ma/tde
🗞️ Newsletter: the-data-entrepreneurs.ck.pag...
Support ❤️
www.buymeacoffee.com/shawhint
Intro - 0:00
What is Fine-tuning? - 0:32
Why Fine-tune - 3:29
3 Ways to Fine-tune - 4:25
Supervised Fine-tuning in 5 Steps - 9:04
3 Options for Parameter Tuning - 10:00
Low-Rank Adaptation (LoRA) - 11:37
Example code: Fine-tuning an LLM with LoRA - 15:40
Load Base Model - 16:02
Data Prep - 17:44
Model Evaluation - 21:49
Fine-tuning with LoRA - 24:10
Fine-tuned Model - 26:50
🔧Fine-tuning: kzhead.info/sun/mKdveMhpn3isoqs/bejne.html 🤖Build a Custom AI Assistant: kzhead.info/sun/Z7Z6pq6sapiNooE/bejne.html 👉Series playlist: kzhead.info/channel/PLz-ep5RbHosU2hnz5ejezwaYpdMutMVB0.html 📰 Read more: towardsdatascience.com/fine-tuning-large-language-models-llms-23473d763b91?sk=fd31e7444cf8f3070d9a843a8218ddad 💻 Example code: github.com/ShawhinT/KZhead-Blog/tree/main/LLMs/fine-tuning -- More Resources [1] Deeplearning.ai Finetuning Large Langauge Models Short Course: www.deeplearning.ai/short-courses/finetuning-large-language-models/ [2] arXiv:2005.14165 [cs.CL] (GPT-3 Paper) [3] arXiv:2303.18223 [cs.CL] (Survey of LLMs) [4] arXiv:2203.02155 [cs.CL] (InstructGPT paper) [5] PEFT: Parameter-Efficient Fine-Tuning of Billion-Scale Models on Low-Resource Hardware: huggingface.co/blog/peft [6] arXiv:2106.09685 [cs.CL] (LoRA paper) [7] Original dataset source - Andrew L. Maas, Raymond E. Daly, Peter T. Pham, Dan Huang, Andrew Y. Ng, and Christopher Potts. 2011. Learning Word Vectors for Sentiment Analysis. In Proceedings of the 49th Annual Meeting of the Association for Computational Linguistics: Human Language Technologies, pages 142-150, Portland, Oregon, USA. Association for Computational Linguistics.
Honestly the most straightforward explanation I've ever watched. Super excellent work Shaw. Thank you. It's so rare to find good communicators like you!
Thanks, glad it was clear 😁
Great video Shaw! It was a good balance between details and concepts. Very unusual to see this so well done. Thank you.
Glad you enjoyed it!
You have explained this so clearly, that even a novice in NLP can understand it.
A very clear and straightforward video explaining finetuning.
Glad it was clear :)
Such a great video. This is the first one I watched from you. You explain everything so nicely, and in my opinion you provided just the right amount of information - not too little, so it doesn't feel superficial and you feel like you've learned something, but not too much, so that you can take what you've learned and read more about it yourself if you're interested. Looking forward to seeing more of your content!
Great to hear! Glad it was clear :)
Your style of conveying information is wonderful. Good luck to you
Great video, I wanted to hear further discussion on mitigation techniques for overfitting. Thanks for making the video!
Just came to this video from HF and I have to say, I love they way you describe this! Thanks for the great video!
Great to hear! Thanks for watching :)
Amazing video Shawhin. It was quite easy to follow and stuff were clearly explained. Thank you so much,
Thanks! I'm glad it was clear and helpful
Wow dude, just you wait, this channel is gonna go viral! You explain everything so clearly, wish you led the courses at my university.
Thanks for the kind words! Maybe one day 😉
excellent simple explanation to the point. Love it !
Thank you for the detailed explaination line by line. Finally a place, I can rely on with working example
Glad it was helpful!
This was one of the best videos on this topic, really nice man, keep going.
Thanks! Glad it was clear :)
Im really gratful for youre work , you really help me when I had no one to ask .
Clear Explanation, Amazing
Excellent way of teaching. Keep doing this kind of good work.
Clear and straightforward to the point, thanks a lot for making this valuable content accessible on ytb💡
Happy to help!
Even though this was high level instruction, it was perfect. I can continue from here. Thanks Shahin jan!
Glad it helped!
Best video i saw. thanks a ton for sharing. glad i found right place
Fantastic video. Thanks for the upload. Keep up the good work, you're awesome 😎
Thanks, I’m glad you liked it 😁
You are the man! No BS, just good useful info
Thanks, glad it was helpful 😁
Thank you for the discussion
I was struggling to understand some details, before this video, thanks a lot
Great to hear. I’m glad it helped!
Very good & simple showcase, thanks
Fantastic explanation.
This is gonna come handy. Thanks for breaking it down
Happy to help!
Great video. Thank you.
Very well explained
Great video, Shawhin!
Thanks, glad you enjoyed it!
i was amazing ....thanks for uploading Shaw
Thanks, happy to help!
Great video! I love good explainations
Thank you, Keep up the good work
Thanks, happy to help!
Very good video and explanation!
Glad it helped!
Excellent walkthrough
🙏
So educative, thanks a lot!
Thanks Shaw, very helpful.
Glad it was helpful!
Excellent..... Thank you for sharing
My pleasure, glad you liked it!
One thing that really standout for me is not using Google Colab for explanation. Explaining all code without scrolling helps the audience better grasp the content as it goes with the flow without waiting for the code to execute and helps the audience to remember where the variables were defined and all. Great approach and thanks for the amazing content!
Thanks, that's good feedback! I'll keep this in mind for future videos.
Well done, even if I already knew all this shit it was really nice to listen to your clear explanation
lol! Glad you enjoyed it :)
HI Shaw, amazing video - very nicely explained! Would be great if you could also do a video (with code examples) for Retrieval Augmented Generation as an alternative to fine-tuning :)
Great suggestion. I have a few follow-up use cases planned out and RAG will definitely be part of it.
@@ShawhinTalebimaybe also how to fine tune openai model too?
Just dropped! kzhead.info/sun/Z7Z6pq6sapiNooE/bejne.html
Thanks for the beautifully explanation!! When you said, for PEFT "we augment the model with additional parameters that are trainable", how do we add these parameters exactly? Do we add a new layer? Also, when we say "%trainable parameters out of total parameters", doesn't that mean that we are updating a certain % of original parameters?
Best video on llm fine tuning. Very concise and informative.
Thanks! Glad you liked it :)
Very clear, thanks!
Thanks Aldo!
Nicely done!
Thanks!
Shaw, terrific job explaining very complicated ideas in an approachable way! One question - are there downsides to combining some of the approaches you mentioned, say, prompt engineering + fine-tuning + RAG to optimize output...how would that compare to using one of the larger OOTB LLMs with hundreds of billions of params?
Great question. The biggest consideration is the tradeoff between cost and performance. On one side you can use an LLM OOTB (e.g. ChatGPT) which costs nothing and has some baseline performance. One the other side you can build a custom system using all the bells and whistles (e.g. fine-tuning, PE, and RAG) which will likely perform much better than ChatGPT but comes at significantly greater cost. Hope that helps!
here, you earned this: 👑
Thanks 🤴
So nice video thank you soooo much!!❤
Happy to help 😁
Nice video !
Really great content. I love your balance of details and overview. It’s made it easy for me as a newcomer who is interested in details. My only criticism/advice is that you edit to remove silence. This is great for minimizing pauses mid sentence. But it would be helpful to have slightly more time at the end of each thought/point. Pausing for that extra 0.25 seconds at the end of a coherent teaching point helps greatly.
Thanks, that's good feedback! I do get a bit heavy-handed with the edits 😅
Random question i how do you edit you audio clips together to make them so seamless because idk where to mate them. And great video by the way 👍
I use iMovie :)
Fantastic job on this overview, as for other videos, I don't see many videos on Inference scaling, i.e requirements for concurrency, latency etc...what are the hardware requirements i.e number of GPUs per systems or number of systems, etc
I'm glad it was helpful :) That's is a great suggestion. I will add it to my list. Thank you!
great video
Thankyou very much it is really very useful .
Happy to help!
My next question after this video would be on how to pack this fine-tuned model into a UI and deploy.
Great question. I discussed how to create a chat interface with Hugging Face + Gradio in a previous video: kzhead.info/sun/ncWnYJufo6GXhZ8/bejne.html
Thank you sooo much❤
You're welcome 😊
I found you in youtube just today. Your presentation style, quality of content is very good. Keep up the great work. I am very passionate about AI technology in general, have been trying to conduct basic trainings to undergraduate college students and would love to connect to collaborate if you are interested. Thank you for doing this!
Thanks for watching! Glad it was clear :) Feel free to set up a call if you like: calendly.com/shawhintalebi
@@ShawhinTalebi Thank you. I will set up some time to connect.
Very helpful! Tysm
Happy to help!
Thanks!
Thanks for the support! Glad it was helpful :)
thank you so much!
Happy to help :)
Greetings! Really nice tutorial! THANK YOU for including Lora! I need to train an Llm on a higher level language we wrote in C++, to produce our code. It's all private infrastructure. Time isnt an issue but I'd like to do it locally on a mac m2 if I can and was considering Lora on a tiny llm. Is this going to be possible?
While I haven't done that myself, that is surely possible. The challenge I've run into is that many open-source models don't work so easily on Mac, but I plan to figure it out and many video about it.
Wow! Amazing make-up! If it wasn't for the voice, I wouldn't believe this is actually David Cross!
Haha, I was wearing jean shorts while filming this 😂
this channel is going to hit 6 figure subscribers at this rate
I hope so 😅
Very good. Very fast and also easy to follow. As far as future content, keep us posted about how to do LoRA on quantized models. How can the future be anything but LoRA on quantized models?!?!?!?
Thanks, glad you liked it. Video coming this quarter on exactly that!
thank you so much
Happy to help!
nice video, thanks😁
Thanks, glad you liked it :)
thanks for sharing this video ....but Is the technique of fine-tuning model for the custom dataset better than the technique of using the RAG system for LLM Apps. or reverse
While the best choice of technique will depend on the specific use case, here's the intuition I've gathered. RAG is great when you want the AI to have access to specific knowledge. Fine-tuning is great when you want the AI to provide responses in a particular format and style. Additionally, the techniques are not mutually exclusive, so they can be combined to potentially provide even better results.
This was beautifully described. I wish you had provided a Linux alternative for the "model.to('mps/cpu'). I have a linux workstation and a p100 gpu. Also, you did not include the means to save your newly trained model. I think most of us students would appreciate knowing how to save the model locally and to huggingface. Thanks for your efforts.
I do fine-tuning on a Linux machine here: kzhead.info/sun/i9Soe6ZrrKWjfps/bejne.html
Hi Shaw - this answered so many questions about specializing an LLM in concise terms, thanks! One question that I'm running up against is physical machine abilities (CPU Speed/Cores, System Memory, GPU cores and memory, and storage speeds. In my case, I have a 32/64 core/thread Epyc CPU on PCIE4.0 MB with 128GB of DDR4 RAM and a PNY/NVIDIA RTX A5000 with 24GB DDR5 VRAM and 8192 CUDA cores dedicated to ML/AI (video is via a separate RTX A2000 GPU). With that info, what should I be looking at as a starting point that will take full advantage of those specs in local mode?
Wow that's a lot of firepower. While I'm less knowledgeable about the ML engineering side of things, I'd suggest checking out DeepSpeed: github.com/microsoft/DeepSpeed. They have several resources on training/running large models efficiently.
@@ShawhinTalebi Thatnks for the pointer. And thinks for all of your output. I've picked up some great information.
Thank you so much for that video. ¿From where did you get TRAINING loss metrics? In the console and in the trainer_state.json, I only see evaluation metrics.
Great question. The training loss is predefined as a property of the base model, so no need to redefine that explicitly.
Hey dude nice video. I think I'll try to find tuned Lamma to detect phrases and subsequently classify tweets - but multiclass classification. Hope it works ,I guess I'll transfer the csv to the prompt you mentioned like alpaca was done and see if it works
Thanks! Sounds like a fun project :)
Hi thanks!! A question for a model in which I have more than 2,000 pdfs. Do you recommend improving the handling of vector databases? When do you recommend fine tunning and when do you recommend vector database
Great question! Generally, fine-tuning and RAG have different strengths. Fine-tuning is great when you want to endow the model with a particular style or to tailor completions for a particular use case, while RAG is good to provide the model with specialized and specific knowledge.
Such a great video ! Wondering how self supervised fine tuning works. Is there any video available on that ?
Thanks! I found this on self-supervised fine-tuning: kzhead.info/sun/iNduiqqoaGiGlI0/bejne.html
Nice Video. I need your help to clarify my doubt. When we do the PEFT based finetuning, the final finetuned model size (in KBs/GBs) will increase by the additional parameters ( base model size + additional parameters size) . In this case base model size will be lesser and final finetuned model size will be more. Deploying the final finetuned model in the edge devices will be more difficult because of the limited edge device resources. Are there any way adapters / LoRA can help in reducing the final finetuned model memory size so that easily we can deploy the final model in the edge devices? Your insights will be helpful. Currently i am working in the vision foundation model deployment in the edge device where i am finding it difficult to deploy because of vision foundation model memory size and inference speed.
Great question. PEFT methods like LoRA only reduce the number of trainable parameters not the total number of parameters. And to your point, the storage requirements actually increase in the case of LoRA! To reduce the final model size, you will need to fine-tune a smaller base model. Hope that helps!
thanks
Would a botpress with a vector kb connected to chatgpt would be enough for Q&A ? When fine tuning starts to be neededvand is there an inexpensive way to do it with no or low code? Thank you!
This depends on the use case. However, taking a quick-and-easy no code approach to start is never a bad idea. It typically gives you a sense of how sophisticated approaches will pan out. Fine-tuning will come into play when the "quick-and-easy" starts to becomes too inconvenient (or expensive) due to the scale of the solution. Hope that helps!
This was a great video. I have one question though. In the LoRA demonstration in your video(at ~14 minutes) you mention this operation (W0 + BA)x = h(x), in this how the sum (W0 + BA) is possible, as W0 has dimentions d*k, and output of operation BA would have the dimentions r*r. This matrix sum is not mathematiaclly possible. So can you elaborate more on this...
Good question! The math works out here because B is d x r and A is r x k, therefore BA will be d x k.
@@ShawhinTalebi 👍
Excellent walk-thru. Thank you, Shaw!I was getting errors on the new model. Switching the device worked for me. # Check if CUDA is available and set the device accordingly device = 'cuda' if torch.cuda.is_available() else 'cpu' model.to(device) # Move the model to the appropriate device (GPU or CPU)
Thanks Manny! That's a good note, I wasn't able to test the code on a non-Mac machine.
Thanks
Welcome :)
Hello! I'm trying to use a similar approach but for a different task. Given a paragraph, I want my model to be able to generate a set of tags associated with it for a specific use case. Not quite sure how the Auto Model would differ here and would love your thoughts on this!
Given you have the structured dataset ready to go, you can use the example code as a jumping off point. You might want to explore alternative base models and fine-tuning approaches. For instance, before using LoRA evaluating the performance of transfer learning alone.
When trying to create a Ai model that generates airticle for a particular niche, is it best to gather airtcle on that niche and Fine-tune it or use open ai knowledge base just giving it some prompts.
Good question. This depends how you are trying to generate the article. If you have a clear structure for how the articles should be written, you can go far with an off-the-shelf model + RAG. However, if the article format is not so rigid (but you have lots of examples), fine-tuning may work best.
Thank you for your great tutorial! What I don't understand is how to use the fine tuned model as an API so we can use it on website. Do you have any tutorial about that?
Great question. I haven't covered that yet, but a common approach is to containerize the model using Docker and equipping it with an API using a library like FastAPI. Then you can host that on a local server or via a cloud provider.
Great video! Is the process for fine tuning a stable diffusion model the same? I think if you make a vid on that itd get a lot of views as well.
I haven't worked with stable diffusion models before, so I don't now, but that would be a great video. Thanks for the suggestion!
Thanks a lot for such a straightforward walkthrough! I tried a similar code for a text generation model, but I keep getting the error 'ValueError: prefetch_factor option could only be specified in multiprocessing. Let num_workers > 0 to enable multiprocessing.' Do you know why this keeps happening? I've even tried changing the torch version, but it's not working.
Not sure that could be. Does the machine have a GPU?
@@ShawhinTalebi Thanks for your help! I figured out the issue was an outdated Linux kernel.
I know you mentioned 1k is a good number of training data for LORA? is it also dependent on model size? If we are using 70b parameter model , will 1k training points be still enough for LORA?
Good question! While this will depend on the use case, 1k is great place to start. I recommend giving it a go and evaluating whether the model performance is acceptable your use case.
Understood. The codes were very helpful. They were not constantly scrolling and panning. But please display the full code and mention the Python version and system configuration, including folders, etc.
Good to hear! All code and env files are available here: github.com/ShawhinT/KZhead-Blog/tree/main/LLMs/fine-tuning
thanks!
Happy to help!
Can I use any open source LLM to train my, for example, healthcare dataset or the LLM should be the one which was pre-trained with healthcare dataset of my interest?
Depends on the use case. If there's an existing healthcare fine-tuned model, why not use that instead of fine-tuning yourself?
Hi, Nice tutorial. I have a question. Is it possible to have more than 1 output in a supervised way? For example: {"input": "ddddddd", "output1":"dddd","eeee", "ffffff", "output2": "xxxx", "zzzzz", etc} Thx
Good question. I'd say it depends on the use case and the meaning of the outputs. However, here are 2 thoughts. 1) concatenate "output1" and "output2" to make "output" e.g. "output1":"dddd","eeee", "ffffff" + "output2": "xxxx", "zzzzz", = "output":"dddd", "eeee", "ffffff", "xxxx", "zzzzz" 2) train 2 models, one for "output1" and another for "output2" Hope that helps!
nice video! is it then at all possible to feed it large amounts of data and make it give correct answers to similar situations as the ones in the database?
Thanks! In principle, yes that is possible with fine-tuning. In practice, this can be a challenge depending on the use case and available data.
@@ShawhinTalebi That was very helpful, thanks man! I will try that out. :)
Thank you. Is there a chance to create own LLM on own computer? A small version? Thank you for information.
It depends what you consider a "Large" Language Model. ~100M parameters is probably the practical limit for (heavy-duty) consumer hardware, at least for now.
@@ShawhinTalebi Maybe there is a small standard LLM available, which is possible to extend/train/finetune with own data. So the first step for the language rule are available. I have now idea if this is possible, that's why I ask, but could be possible.
Can you recommend any course where i can learn to build llm from scratch and fine-tune in depth
Paul Iusztin has some good content on that. Hands-on-llms: github.com/iusztinpaul/hands-on-llms More resources: www.pauliusztin.me/
How to control the % of params that are being trained? Where are we specifying this? Also can you pls tell me how to choose r? What are these r values: 2,4,8 etc?
When using LoRA you control the number of trainable parameters via the r value and target modules. These are both specified at 24:10, where r=4 and only the query layers are augmented. As for choosing r this depends on your use case. Small r means less parameters but (generally) worse performance, while large r means more parameters and better performance.
Can you show us how to do transfer learning for open source llms, and why that should be the first step for fine tuning a model? Is it more efficient way of finetuning?
Great suggestion! Next video will touch on this by covering how to fine-tune open-source LLMs with QLoRA.
This feature could already be available on KZhead for creators. Perhaps, you could refine a chatbot that can automatically respond to comments using Gemini. It could even learn to respond based on your videos, eliminating the need for you to upload anything or messing with fine-tuning.
It is to some extent, as we get response recommendations in the creator studio. Using multimodal models might takes this to the next level!
is there a way that distilbert or any other LLM can be trained for QA using dataset that has only text field without any label? I'm trying to trian the LLM for QA but my dataset has only text field without any labels or questions and answers.
What does your text field consist of? Does it include questions or answers?
@@ShawhinTalebi no only raw text, you may refer to any dataset of hf website that has only (text field & ID) so I'm trying to fine-tune the model on the Arabic dataset which is only raw text. Appreciate it, if you can make a video on: 1. how to fine-tune the model on languages other than English (because the model is originaly trained on English) 2. how to fien-tune the model with data that only has text and use the model for QA 3. Will the model that is not trained on English originally require pre-training and then fine-tuning
If you only have raw text, you will likely need to due data preprocessing to generate input-output pairs for fine-tuning. Thanks for the suggestions!!
nice
Thanks
Is there any limitation to the GPU memory? I am just a student with only a 3050 GPU with only 4GB memory
Great question. While it may take some time, the example here should run on a CPU, so I suspect it should run fine with your GPU. Give it a try and let me know how it goes.
This is incredible, thank you for the clear tutorial. Please subscribe to this channel. One question: Can we apply LoRA to finetune models used in image classification or any computer vision problems? Links to read or a short tutorial would be helpful.
Thanks, glad it was clear! Yes! LoRA is not specific to language models. Here is a guide on image classification using LoRA from HF: huggingface.co/docs/peft/task_guides/image_classification_lora
@@ShawhinTalebi Thank you for the link.
Did you do this fine tuning on CPU or GPU, can you provide details? Thanks
I have a Mac M1 which uses unified memory (i.e. GPU and CPU are one).