Update .github/workflows for pytest

Author: MohsenDehghankar

.github/workflows/tests.yml

@@ -28,12 +28,13 @@ jobs:
       - name: Install Python dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install -r requirements.txt
+          python -m pip install --upgrade -r requirements.txt
+          python -m pip install -e .
 
       - name: Rebuild CPU kernel libraries
         run: |
           make -C kernels/bit_1/cpu clean all
           make -C kernels/bit_1_58/cpu clean all
 
       - name: Run test suite
-        run: pytest -q tests
+        run: python -m pytest -q tests

conftest.py

@@ -0,0 +1,4 @@
+import sys
+import os
+
+sys.path.insert(0, os.path.dirname(__file__))

integrations/hf/model_infer.py

@@ -18,6 +18,7 @@
 import copy
 import json
 import sys
+from contextlib import contextmanager
 from pathlib import Path
 from typing import Any
 
@@ -27,9 +28,15 @@
 from safetensors import safe_open
 from safetensors.torch import load_file as safetensors_load
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
-from transformers.modeling_utils import no_init_weights
 from transformers.models.auto.configuration_auto import CONFIG_MAPPING
 
+try:
+    from transformers.modeling_utils import no_init_weights
+except ImportError:
+    @contextmanager
+    def no_init_weights():
+        yield
+
 _PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent
 if str(_PROJECT_ROOT) not in sys.path:
     sys.path.insert(0, str(_PROJECT_ROOT))