Skip to content

Commit 87b3714

Browse files
authored
Update neuralmagic --> vllm-project (#495)
1 parent 1ef2723 commit 87b3714

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

.github/workflows/report.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ jobs:
108108
"hostname": "${{ secrets.REPORTPORTAL_HOST }}",
109109
"project": "compressed-tensors",
110110
"name": "${{ inputs.run_name }}",
111-
"description": "GitHub run: https://github.com/neuralmagic/compressed-tensors/actions/runs/${{ inputs.run_id }}",
111+
"description": "GitHub run: https://github.com/vllm-project/compressed-tensors/actions/runs/${{ inputs.run_id }}",
112112
"attributes": [
113113
{"key": "wheel", "value": "${{ inputs.wheel }}"},
114114
{"key": "gitref", "value": "${{ inputs.gitref }}"},

README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ pip install --pre compressed-tensors
3737
### From Source
3838

3939
```bash
40-
git clone https://github.com/neuralmagic/compressed-tensors
40+
git clone https://github.com/vllm-project/compressed-tensors
4141
cd compressed-tensors
4242
pip install -e .
4343
```
@@ -78,7 +78,7 @@ We can apply bitmask compression to a whole model. For more detailed example see
7878
from compressed_tensors import save_compressed_model, load_compressed, BitmaskConfig
7979
from transformers import AutoModelForCausalLM
8080

81-
model_name = "neuralmagic/llama2.c-stories110M-pruned50"
81+
model_name = "RedHatAI/llama2.c-stories110M-pruned50"
8282
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto")
8383

8484
original_state_dict = model.state_dict()
@@ -92,7 +92,7 @@ save_compressed_model(model, "compressed_model.safetensors", compression_format=
9292
state_dict = dict(load_compressed("compressed_model.safetensors", compression_config))
9393
```
9494

95-
For more in-depth tutorial on bitmask compression, refer to the [notebook](https://github.com/neuralmagic/compressed-tensors/blob/d707c5b84bc3fef164aebdcd97cb6eaa571982f8/examples/bitmask_compression.ipynb).
95+
For more in-depth tutorial on bitmask compression, refer to the [notebook](https://github.com/vllm-project/compressed-tensors/blob/d707c5b84bc3fef164aebdcd97cb6eaa571982f8/examples/bitmask_compression.ipynb).
9696

9797

9898
## Saving a Compressed Model with PTQ

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def _setup_extras() -> Dict:
109109
description="Library for utilization of compressed safetensors of neural network models",
110110
long_description=_setup_long_description()[0],
111111
long_description_content_type=_setup_long_description()[1],
112-
url="https://github.com/neuralmagic/compressed-tensors",
112+
url="https://github.com/vllm-project/compressed-tensors",
113113
extras_require=_setup_extras(),
114114
install_requires=_setup_install_requires(),
115115
package_dir={"": "src"},

0 commit comments

Comments
 (0)