g

Author: xadupre

.github/workflows/ci.yml

@@ -80,6 +80,7 @@ jobs:
         run: |
           export PYTHONPATH=.
           python _doc/examples/plot_export_tiny_llm.py
+        continue-on-error: true
 
       - name: tiny-llm bypass
         run: |

_unittests/ut_torch_models/test_hghub_model.py

@@ -81,6 +81,15 @@ def test_get_untrained_model_with_inputs_beit(self):
         # different expected value for different version of transformers
         self.assertIn((data["size"], data["n_weights"]), [(111448, 27862)])
 
+    @hide_stdout()
+    def test_get_untrained_model_with_inputs_codellama(self):
+        mid = "codellama/CodeLlama-7b-Python-hf"
+        data = get_untrained_model_with_inputs(mid, verbose=1)
+        model, inputs = data["model"], data["inputs"]
+        model(**inputs)
+        # different expected value for different version of transformers
+        self.assertIn((data["size"], data["n_weights"]), [(410532864, 102633216)])
+
     @hide_stdout()
     @long_test()
     def test_get_untrained_model_Ltesting_models(self):