Unverified Commit de125d6d authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.compressed-tensors: 0.10.2 -> 0.11.0 (#436310)

parents 445f1846 ee57b87f

pkgs/development/python-modules/compressed-tensors/default.nix

+27 −6
Original line number Diff line number Diff line 
{

  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 = [