From 514293b332c15638424ef0b9e2d0e13b794bb914 Mon Sep 17 00:00:00 2001 From: vkovinicTT Date: Fri, 7 Mar 2025 09:25:21 +0000 Subject: [PATCH] skipped falcon 3B model due to insufficient DRAM --- forge/test/models/pytorch/text/falcon/test_falcon.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/forge/test/models/pytorch/text/falcon/test_falcon.py b/forge/test/models/pytorch/text/falcon/test_falcon.py index 5ef61757c..3791d2932 100644 --- a/forge/test/models/pytorch/text/falcon/test_falcon.py +++ b/forge/test/models/pytorch/text/falcon/test_falcon.py @@ -64,6 +64,9 @@ def test_falcon_3(record_forge_property, variant): if variant == "tiiuae/Falcon3-Mamba-7B-Base" or variant == "tiiuae/Falcon3-7B-Base": pytest.skip("Insufficient host DRAM to run this model (requires a bit more than 36 GB)") + if variant == "tiiuae/Falcon3-3B-Base": + pytest.skip("Insufficient host DRAM to run this model (requires a bit more than 25 GB)") + # Build Module Name module_name = build_module_name( framework=Framework.PYTORCH, model="falcon3", variant=variant, task=Task.CAUSAL_LM, source=Source.HUGGINGFACE