From c56ec07a9ac2bebdf868e8ce052299b3ec59ca4b Mon Sep 17 00:00:00 2001
From: ibrahimkhadraoui <ibrahim.khadraoui@tii.ae>
Date: Mon, 7 Jul 2025 10:34:46 +0400
Subject: [PATCH] read arch from gguf.MODEL_ARCH

---
 convert_hf_to_gguf.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py
index 96cc8d28c7..89bfc33c8c 100755
--- a/convert_hf_to_gguf.py
+++ b/convert_hf_to_gguf.py
@@ -4909,8 +4909,7 @@ class Mamba2Model(TextModel):
         # Fail early for models which don't have a block expansion factor of 2
         # TODO: does this really matter?
         # skip the assertion for FalconH1 Model
-        architectures = self.hparams.get("architectures")
-        if architectures is None or architectures[0] != "FalconH1ForCausalLM":
+        if self.model_arch != gguf.MODEL_ARCH.FALCON_H1:
             assert d_inner == 2 * d_model
             assert d_inner % head_dim == 0