Add CPU test for falcon3 Model

forge/test/models/pytorch/text/falcon/test_falcon.py

@@ -47,3 +47,38 @@ def forward(self, input_ids, attention_mask):
 
     # Model Verification
     verify(inputs, framework_model, compiled_model)
+
+
+variants = [
+    "tiiuae/Falcon3-1B-Base",
+    "tiiuae/Falcon3-3B-Base",
+    "tiiuae/Falcon3-7B-Base",
+    "tiiuae/Falcon3-Mamba-7B-Base",
+]
+
+
+@pytest.mark.nightly
+@pytest.mark.parametrize("variant", variants)
+def test_falcon_3(record_forge_property, variant):
+
+    # Build Module Name
+    module_name = build_module_name(
+        framework=Framework.PYTORCH, model="falcon", variant=variant, task=Task.CAUSAL_LM, source=Source.HUGGINGFACE
+    )
+
+    # Record Forge Property
+    record_forge_property("model_name", module_name)
+
+    tokenizer = AutoTokenizer.from_pretrained(variant)
+    model = AutoModelForCausalLM.from_pretrained(variant)
+    model.config.use_cache = False
+    model.config.return_dict = False
+
+    input_text = "Write a function to calculate the factorial of a number"
+    input_data = tokenizer.encode(input_text, return_tensors="pt")
+
+    # Forge compile framework model
+    compiled_model = forge.compile(model, sample_inputs=input_data, module_name=module_name)
+
+    # Model Verification
+    verify([input_data], model, compiled_model)