Skip to content

Commit 66fe9ff

Browse files
committed
convert : add Llama4ForCausalLM
1 parent 8ff2060 commit 66fe9ff

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2393,7 +2393,10 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
23932393
return [] # skip other tensors
23942394

23952395

2396-
@ModelBase.register("Llama4ForConditionalGeneration")
2396+
@ModelBase.register(
2397+
"Llama4ForConditionalGeneration",
2398+
"Llama4ForCausalLM",
2399+
)
23972400
class Llama4Model(LlamaModel):
23982401
model_arch = gguf.MODEL_ARCH.LLAMA4
23992402
undo_permute = False

0 commit comments

Comments
 (0)