Skip to content
This repository has been archived by the owner on Oct 11, 2024. It is now read-only.

Commit

Permalink
Mark files that have been modified by Neural Magic (#70)
Browse files Browse the repository at this point in the history
SUMMARY:
for Apache 4(b) -- "You must cause any modified files to carry prominent
notices stating that You changed the files"
https://www.apache.org/licenses/LICENSE-2.0 

TEST PLAN:
GHA
  • Loading branch information
tlrmchlsmth authored Feb 28, 2024
1 parent 3e1d327 commit 13787e6
Show file tree
Hide file tree
Showing 20 changed files with 36 additions and 0 deletions.
2 changes: 2 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

# Byte-compiled / optimized / DLL files
__pycache__/
*.py[cod]
Expand Down
2 changes: 2 additions & 0 deletions benchmarks/backend_request_func.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import json
import os
import time
Expand Down
2 changes: 2 additions & 0 deletions csrc/ops.h
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
// This file has been modified by Neural Magic

#pragma once

#include <torch/extension.h>
Expand Down
2 changes: 2 additions & 0 deletions csrc/pybind.cpp
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
// This file has been modified by Neural Magic

#include "cache.h"
#include "cuda_utils.h"
#include "ops.h"
Expand Down
2 changes: 2 additions & 0 deletions requirements-dev.txt
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

# formatting
yapf==0.32.0
toml==0.10.2
Expand Down
2 changes: 2 additions & 0 deletions setup.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import contextlib
import io
import os
Expand Down
2 changes: 2 additions & 0 deletions tests/conftest.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import os
from typing import List, Optional, Tuple

Expand Down
2 changes: 2 additions & 0 deletions tests/distributed/test_custom_all_reduce.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import random

import os
Expand Down
2 changes: 2 additions & 0 deletions tests/entrypoints/test_openai_server.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import os
import subprocess
import time
Expand Down
1 change: 1 addition & 0 deletions tests/models/test_mistral.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# This file has been modified by Neural Magic
"""Compare the outputs of HF and vLLM for Mistral models using greedy sampling.
Run `pytest tests/models/test_mistral.py --forked`.
Expand Down
1 change: 1 addition & 0 deletions tests/models/test_models.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# This file has been modified by Neural Magic
"""Compare the outputs of HF and vLLM when using greedy sampling.
Run `pytest tests/models/test_models.py --forked`.
Expand Down
2 changes: 2 additions & 0 deletions vllm/config.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

from typing import Optional, Union, ClassVar
from dataclasses import dataclass
import os
Expand Down
2 changes: 2 additions & 0 deletions vllm/engine/arg_utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import argparse
import dataclasses
from dataclasses import dataclass
Expand Down
2 changes: 2 additions & 0 deletions vllm/engine/llm_engine.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import copy
from collections import defaultdict
import os
Expand Down
2 changes: 2 additions & 0 deletions vllm/entrypoints/llm.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

from typing import List, Optional, Union

from tqdm import tqdm
Expand Down
2 changes: 2 additions & 0 deletions vllm/model_executor/layers/linear.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional

Expand Down
2 changes: 2 additions & 0 deletions vllm/model_executor/layers/quantization/__init__.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

from typing import Type

from vllm.model_executor.layers.quantization.base_config import QuantizationConfig
Expand Down
1 change: 1 addition & 0 deletions vllm/model_executor/model_loader.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# This file has been modified by Neural Magic
"""Utilities for selecting and loading models."""
import contextlib
from typing import Optional, Type
Expand Down
1 change: 1 addition & 0 deletions vllm/model_executor/weight_utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# This file has been modified by Neural Magic
"""Utilities for downloading and initializing model weights."""
import filelock
import glob
Expand Down
2 changes: 2 additions & 0 deletions vllm/test_utils.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
# This file has been modified by Neural Magic

import ray

from vllm.config import ParallelConfig
Expand Down

0 comments on commit 13787e6

Please sign in to comment.