Fix architecture detection

This commit is contained in:
turboderp
2024-02-16 01:52:26 +01:00
parent cedeb616ce
commit 5967a29eb4

View File

@@ -146,7 +146,7 @@ class ExLlamaV2Config:
expect_keys += \
expect_keys_llama
if "Qwen2ForCausalLM" in read_config["architectures"]:
elif "Qwen2ForCausalLM" in read_config["architectures"]:
self.architecture = "Qwen2"
layer_keys += \
layer_keys_llama_norms + \