Commit 60d3b6ae authored by Fabian Affolter's avatar Fabian Affolter Committed by github-actions[bot]
Browse files

python312Packages.curated-transformers: 0.1.1 -> 2.0.1

parent a60bad2b
Loading
Loading
Loading
Loading
+13 −3
Original line number Diff line number Diff line
{
  lib,
  buildPythonPackage,
  catalogue,
  curated-tokenizers,
  fetchFromGitHub,
  huggingface-hub,
  setuptools,
  tokenizers,
  torch,
}:

buildPythonPackage rec {
  pname = "curated-transformers";
  version = "0.1.1";
  version = "2.0.1";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "explosion";
    repo = "curated-transformers";
    tag = "v${version}";
    hash = "sha256-QhJZnQIa9TilwdQCUlxnQCEc6Suj669cht6WHUAr/Gw=";
    hash = "sha256-2sedBVpwCppviWix+d3tJFTrLBe+2IBlWnCKgV6MucA=";
  };

  build-system = [ setuptools ];

  dependencies = [ torch ];
  dependencies = [
    catalogue
    curated-tokenizers
    huggingface-hub
    tokenizers
    torch
  ];

  # Unit tests are hard to use, since most tests rely on downloading
  # models from Hugging Face Hub.