Merge branch 'main' into bdellabe/awq-w4a8

Author: brian-dellabetta

.coveragerc

@@ -0,0 +1,2 @@
+[run]
+patch = subprocess

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,47 @@
+name: 🐛 Bug report
+description: Raise an issue here if you find a bug.
+labels: bug
+title: "[Bug]: "
+
+body:
+- type: markdown
+  attributes:
+    value: >
+      #### Before submitting an issue, please make sure the issue hasn't been already addressed by searching through [the existing and past issues](https://github.com/vllm-project/llm-compressor/issues?q=is%3Aissue+sort%3Acreated-desc+).
+
+      #### ⚠️ For any issues related vLLM which are not related to quantization or compressed models, please create an issue in [vllm-project/vllm](https://github.com/vllm-project/vllm/issues).
+- type: textarea
+  attributes:
+    label: ⚙️ Your current environment
+    description: |
+      Please run the following and paste the output below.
+      ```bash
+      wget https://raw.githubusercontent.com/vllm-project/llm-compressor/main/tools/collect_env.py
+      # For security purposes, please feel free to check the contents of collect_env.py before running it.
+      python collect_env.py
+      ```
+    value: |
+      <details>
+      <summary>The output of <code>python collect_env.py</code></summary>
+
+      ```text
+      Your output of `python collect_env.py` here
+      ```
+
+      </details>
+  validations:
+    required: true
+- type: textarea
+  attributes:
+    label: 🐛 Describe the bug
+    description: |
+      Please provide a clear and concise description of what the bug is.
+  validations:
+    required: true
+- type: textarea
+  attributes:
+    label: 🛠️ Steps to reproduce
+    description: |
+      If applicable, please describe any steps required to reproduce. If you can share an applicable huggingface model stub, please do so here.
+  validations:
+    required: false

.github/ISSUE_TEMPLATE/bug_report.yaml

@@ -2,11 +2,9 @@ name: Check Markdown links
 
 on:
   push:
-    branches:
-      - main
+    branches: [ main, 'release/*' ]
   pull_request:
-    branches:
-      - main
+    branches: [ main, 'release/*' ]
 
   # Allows you to run this workflow manually from the Actions tab
   workflow_dispatch:

.github/workflows/linkcheck.yml

@@ -1,13 +1,9 @@
 name: Quality Checks
 on:
   push:
-    branches:
-      - main
-      - 'release/*'
+    branches: [ main , 'release/*' ]
   pull_request:
-    branches:
-      - main
-      - 'release/*'
+    branches: [ main, 'release/*' ]
 jobs:
   quality-check:
     runs-on: ubuntu-22.04

.github/workflows/quality-check.yaml

@@ -1,7 +1,7 @@
 name: PR Reminder Comment Bot
 on:
   pull_request_target:
-    branches: [main]
+    branches: [ main, 'release/*' ]
     types: [opened]
 
 jobs:

.github/workflows/set-comment.yaml

@@ -1,10 +1,10 @@
 name: Test Checks (Transformers)
 on:
   pull_request:
-    branches: [ main ]
+    branches: [ main, 'release/*' ]
     types: [ labeled, synchronize ]
   push:
-    branches: [ main ]
+    branches: [ main, 'release/*' ]
   workflow_dispatch:
     inputs:
       code_coverage:

.github/workflows/test-check-transformers.yaml

@@ -1,9 +1,9 @@
 name: Test Checks (Base/PyTorch)
 on:
   pull_request:
-    branches:
-      - main
+    branches: [ main, 'release/*' ]
   push:
+    branches: [ main, 'release/*' ]
   workflow_dispatch:
     inputs:
       code_coverage:
@@ -115,59 +115,9 @@ jobs:
         run: |
           coverage report --data-file="$COVERAGE_FILE" --skip-empty --format="markdown" > "$GITHUB_STEP_SUMMARY"
 
-  compat-pytorch-1_9-pytorch-tests:
-    runs-on: ubuntu-22.04
-    env:
-      COVERAGE_FILE: ".coverage.compat-pytorch-1.9"
-    steps:
-      - uses: actions/setup-python@v5
-        with:
-          python-version: '3.10'
-      - uses: actions/checkout@v4
-        with:
-          fetch-depth: 0
-          fetch-tags: true
-      - name: "⚙️ Install dependencies"
-        run: pip3 install -U pip setuptools && pip3 install .[dev]
-      - uses: actions/checkout@v4
-        with:
-          repository: "neuralmagic/compressed-tensors"
-          path: "compressed-tensors"
-          fetch-depth: 0
-          fetch-tags: true
-      - name: "⚙️ Install compressed-tensors dependencies"
-        run: |
-          pip3 uninstall -y compressed-tensors
-          export GIT_CEILING_DIRECTORIES="$(pwd)"
-          cd compressed-tensors
-          BUILD_TYPE=nightly pip3 install .
-      - name: "Clean compressed-tensors directory"
-        run: rm -r compressed-tensors/
-      - name: "⚙️ Prepare code coverage"
-        if: inputs.code_coverage
-        uses: ./.github/actions/prepare-code-coverage
-      - name: "🔬 Running pytorch tests"
-        run: |
-          pytest -v tests/llmcompressor/pytorch
-      - name: "Upload coverage report"
-        if: (success() || failure()) && inputs.code_coverage
-        uses: actions/upload-artifact@v4
-        with:
-          name: compat-pytorch-tests-coverage-results
-          path: |
-            .coverage*
-            coverage-html
-            coverage.json
-          include-hidden-files: true
-          retention-days: 5
-      - name: "Report coverage"
-        if: (success() || failure()) && inputs.code_coverage
-        run: |
-          coverage report --data-file="$COVERAGE_FILE" --skip-empty --format="markdown" > "$GITHUB_STEP_SUMMARY"
-
   combine-coverage:
     runs-on: ubuntu-22.04
-    needs: [base-tests, pytorch-tests, compat-pytorch-1_9-pytorch-tests]
+    needs: [base-tests, pytorch-tests]
     if: (success() || failure()) && inputs.code_coverage
     steps:
       - name: "Checkout llm-compressor"

.github/workflows/test-check.yaml

@@ -93,9 +93,6 @@ instance/
 # Scrapy stuff:
 .scrapy
 
-# Sphinx documentation
-docs/_build/
-
 # PyBuilder
 target/
 
@@ -129,6 +126,7 @@ venv.bak/
 
 # mkdocs documentation
 /site
+docs/.cache/
 
 # mypy
 .mypy_cache/

.gitignore

@@ -25,6 +25,7 @@ endif
 quality:
 	@echo "Running python quality checks";
 	ruff check $(CHECKDIRS);
+	ruff format --check $(CHECKDIRS);
 	isort --check-only $(CHECKDIRS);
 	flake8 $(CHECKDIRS) --max-line-length 88 --extend-ignore E203,W605;