Co-authored-by: dinhhuy <redacted>
"MistralForCausalLM",
"MixtralForCausalLM",
"VLlama3ForCausalLM",
- "LlavaForConditionalGeneration")
+ "LlavaForConditionalGeneration",
+ "LlamaModel")
class LlamaModel(TextModel):
model_arch = gguf.MODEL_ARCH.LLAMA
undo_permute = True
if is_vision_tensor:
return [] # skip vision tensors
+ elif self.hf_arch == "LlamaModel":
+ name = "model." + name
elif name.startswith("model.text_model"):
name = name.replace("text_model.", "") # for SmolVLM
elif name.startswith("language_model."):