Merge pull request #2017 from gs-olive/bert_ci_fix

fix: Repair input range on BERT inputs for CI

Author: peri044

tests/py/models/test_models.py

@@ -86,7 +86,7 @@ def test_efficientnet_b0(self):
 
     def test_bert_base_uncased(self):
         self.model = cm.BertModule().cuda()
-        self.input = torch.randint(0, 5, (1, 14), dtype=torch.int32).to("cuda")
+        self.input = torch.randint(0, 2, (1, 14), dtype=torch.int32).to("cuda")
 
         compile_spec = {
             "inputs": [