python3Packages.compressed-tensors: 0.10.2 -> 0.11.0 (#436310)
This commit is contained in:
@@ -1,12 +1,19 @@
|
||||
{
|
||||
lib,
|
||||
stdenv,
|
||||
buildPythonPackage,
|
||||
fetchFromGitHub,
|
||||
|
||||
# build-system
|
||||
setuptools,
|
||||
setuptools-scm,
|
||||
|
||||
# dependencies
|
||||
frozendict,
|
||||
pydantic,
|
||||
torch,
|
||||
transformers,
|
||||
|
||||
# tests
|
||||
nbconvert,
|
||||
nbformat,
|
||||
pytestCheckHook,
|
||||
@@ -14,7 +21,7 @@
|
||||
|
||||
buildPythonPackage rec {
|
||||
pname = "compressed-tensors";
|
||||
version = "0.10.2";
|
||||
version = "0.11.0";
|
||||
pyproject = true;
|
||||
|
||||
# Release on PyPI is missing the `utils` directory, which `setup.py` wants to import
|
||||
@@ -22,12 +29,21 @@ buildPythonPackage rec {
|
||||
owner = "neuralmagic";
|
||||
repo = "compressed-tensors";
|
||||
tag = version;
|
||||
hash = "sha256-BJsMyCs+rupt5+i5JlO7oY08Udc8hI3ZnMiN+8ja0mc=";
|
||||
hash = "sha256-sSXn4/N/Pn+wOCY1Z0ziqFxfMRvRA1c90jPOBe+SwZw=";
|
||||
};
|
||||
|
||||
build-system = [ setuptools ];
|
||||
postPatch = ''
|
||||
substituteInPlace pyproject.toml \
|
||||
--replace-fail "setuptools_scm==8.2.0" "setuptools_scm"
|
||||
'';
|
||||
|
||||
build-system = [
|
||||
setuptools
|
||||
setuptools-scm
|
||||
];
|
||||
|
||||
dependencies = [
|
||||
frozendict
|
||||
pydantic
|
||||
torch
|
||||
transformers
|
||||
@@ -45,12 +61,17 @@ buildPythonPackage rec {
|
||||
|
||||
disabledTests = [
|
||||
# these try to download models from HF Hub
|
||||
"test_apply_tinyllama_dynamic_activations"
|
||||
"test_compress_model"
|
||||
"test_compress_model_meta"
|
||||
"test_compressed_linear_from_linear_usage"
|
||||
"test_decompress_model"
|
||||
"test_get_observer_token_count"
|
||||
"test_kv_cache_quantization"
|
||||
"test_target_prioritization"
|
||||
"test_load_compressed_sharded"
|
||||
"test_model_forward_pass"
|
||||
"test_save_compressed_model"
|
||||
"test_apply_tinyllama_dynamic_activations"
|
||||
"test_target_prioritization"
|
||||
];
|
||||
|
||||
disabledTestPaths = [
|
||||
|
||||
Reference in New Issue
Block a user