open_llama icon indicating copy to clipboard operation
open_llama copied to clipboard

Thanks! Does it fit in Free Colab?

Open amrrs opened this issue 2 years ago • 3 comments

Hey Thanks very much for the effort. Looking at the released model size, Wondering if it can fit in Free Colab?

amrrs avatar May 02 '23 20:05 amrrs

The model architecture and size is the same as LLaMA 7B, so if the original LLaMA 7B fits, our model should fit as well. If not, we are also working on a smaller LLaMA 3B model, which we will release a preview checkpoint soon.

young-geng avatar May 03 '23 00:05 young-geng

@amrrs if you manage to get it working in a Colab i'd be very keen too!

swyxio avatar May 03 '23 01:05 swyxio

I made a video on how to run it in Google Colab (it fits in T4 GPU): https://www.youtube.com/watch?v=1NOPciKuQb8

Hope that helps!

curiousily avatar May 06 '23 13:05 curiousily

We've just released a checkpoint for our 3B model and that should definitely fit in colab.

young-geng avatar May 17 '23 20:05 young-geng