LLaVA icon indicating copy to clipboard operation
LLaVA copied to clipboard

[Question] How to use the pretrain checkpoint

Open hellangleZ opened this issue 9 months ago • 3 comments

Question

I only found that there are some file like:

image

How can I merge them to base model , or something I should do, any help is very appreciated

hellangleZ avatar Apr 29 '24 02:04 hellangleZ

I have the same question...

feizhao19 avatar Apr 29 '24 03:04 feizhao19

You can use the load_pretrained_model function. Check here.

You need to:

  1. pass a model-base argument, which is the base language model you used for your llava model.
  2. pass a model-path argument, which is where your pretrained weights' folder is.

This way, you can load the pretrained weight into a base llava model.

baichuanzhou avatar Apr 29 '24 08:04 baichuanzhou

Is it possible to use the pretrained checkpoints with SGLang?

lukashelff avatar Apr 29 '24 09:04 lukashelff