News for hackers. Stuff that matters.
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by fbodz on Monday May 15, 2023 @06:18PM (#35955266)
Has anyone figured out a way to fine tune this with 24gb of vram? I have tried with deepspeed etc but no luck. Seems to be just out of reach for fine tuning requiring 26gb.
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by csdvrx on Monday May 15, 2023 @06:55PM (#35955625)
Have you tried quantization? It's often a cheap and simple way to reduce the VRAM requirements.

What hardware are you using? (CPU,RAM,GPU,VRAM)

Have you considered using llama.cpp for a mixed CPU+GPU use (if you have enough RAM)

Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by fbodz on Monday May 15, 2023 @07:12PM (#35955796)
Yeah I am using the default training script with int8 quantisation. It uses peft with lora but this still requires 26gb
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by freeqaz on Monday May 15, 2023 @07:57PM (#35956170)
Is it possible to offload some layers to CPU and still train in a reasonable amount of time?
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by generalizations on Monday May 15, 2023 @08:15PM (#35956297)
There’s also that pruning tool that was on hn in the last couple weeks. It seemed to work really well on the larger models, and could reduce size by 30-50%
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by int_19h on Monday May 15, 2023 @09:28PM (#35956761)
I'm not sure about this model specifically, but training with 4-bit quantization has been a thing with LLaMA for a while now, although the setup involves manual hacks of various libraries.
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by nl on Tuesday May 16, 2023 @02:48AM (#35958456)
You probably don't want to fine-tune a quantized model. They are fine for inference but not great for training.
Re:StarCoder and StarCoderBase: 15.5B parameter models with ... by mirekrusin on Tuesday May 16, 2023 @01:52AM (#35958142)
People should be training model sizes that fit-and-fill consumer GPUs, ie:

2x 24G - for dual GPU ~ 28B model

1x 24G ~ 14B model

etc.