mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-12-22 21:53:32 +00:00
161 lines
3.6 KiB
Nix
161 lines
3.6 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
pythonOlder,
|
|
fetchFromGitHub,
|
|
writeText,
|
|
setuptools,
|
|
filelock,
|
|
huggingface-hub,
|
|
importlib-metadata,
|
|
numpy,
|
|
pillow,
|
|
regex,
|
|
requests,
|
|
safetensors,
|
|
# optional dependencies
|
|
accelerate,
|
|
datasets,
|
|
flax,
|
|
jax,
|
|
jaxlib,
|
|
jinja2,
|
|
peft,
|
|
protobuf,
|
|
tensorboard,
|
|
torch,
|
|
# test dependencies
|
|
parameterized,
|
|
pytest-timeout,
|
|
pytest-xdist,
|
|
pytestCheckHook,
|
|
requests-mock,
|
|
scipy,
|
|
sentencepiece,
|
|
torchsde,
|
|
transformers,
|
|
pythonAtLeast,
|
|
diffusers,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "diffusers";
|
|
version = "0.30.3";
|
|
pyproject = true;
|
|
|
|
disabled = pythonOlder "3.8";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "huggingface";
|
|
repo = "diffusers";
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-/3lHJdsNblKb6xX03OluSCApMK3EXJbRLboBk8CjobE=";
|
|
};
|
|
|
|
build-system = [ setuptools ];
|
|
|
|
dependencies = [
|
|
filelock
|
|
huggingface-hub
|
|
importlib-metadata
|
|
numpy
|
|
pillow
|
|
regex
|
|
requests
|
|
safetensors
|
|
];
|
|
|
|
optional-dependencies = {
|
|
flax = [
|
|
flax
|
|
jax
|
|
jaxlib
|
|
];
|
|
torch = [
|
|
accelerate
|
|
torch
|
|
];
|
|
training = [
|
|
accelerate
|
|
datasets
|
|
jinja2
|
|
peft
|
|
protobuf
|
|
tensorboard
|
|
];
|
|
};
|
|
|
|
pythonImportsCheck = [ "diffusers" ];
|
|
|
|
# it takes a few hours
|
|
doCheck = false;
|
|
|
|
passthru.tests.pytest = diffusers.overridePythonAttrs { doCheck = true; };
|
|
|
|
nativeCheckInputs = [
|
|
parameterized
|
|
pytest-timeout
|
|
pytest-xdist
|
|
pytestCheckHook
|
|
requests-mock
|
|
scipy
|
|
sentencepiece
|
|
torchsde
|
|
transformers
|
|
] ++ optional-dependencies.torch;
|
|
|
|
preCheck =
|
|
let
|
|
# This pytest hook mocks and catches attempts at accessing the network
|
|
# tests that try to access the network will raise, get caught, be marked as skipped and tagged as xfailed.
|
|
# cf. python3Packages.shap
|
|
conftestSkipNetworkErrors = writeText "conftest.py" ''
|
|
from _pytest.runner import pytest_runtest_makereport as orig_pytest_runtest_makereport
|
|
import urllib3
|
|
|
|
class NetworkAccessDeniedError(RuntimeError): pass
|
|
def deny_network_access(*a, **kw):
|
|
raise NetworkAccessDeniedError
|
|
|
|
urllib3.connection.HTTPSConnection._new_conn = deny_network_access
|
|
|
|
def pytest_runtest_makereport(item, call):
|
|
tr = orig_pytest_runtest_makereport(item, call)
|
|
if call.excinfo is not None and call.excinfo.type is NetworkAccessDeniedError:
|
|
tr.outcome = 'skipped'
|
|
tr.wasxfail = "reason: Requires network access."
|
|
return tr
|
|
'';
|
|
in
|
|
''
|
|
export HOME=$TMPDIR
|
|
cat ${conftestSkipNetworkErrors} >> tests/conftest.py
|
|
'';
|
|
|
|
pytestFlagsArray = [ "tests/" ];
|
|
|
|
disabledTests =
|
|
[
|
|
# depends on current working directory
|
|
"test_deprecate_stacklevel"
|
|
# fails due to precision of floating point numbers
|
|
"test_model_cpu_offload_forward_pass"
|
|
# tries to run ruff which we have intentionally removed from nativeCheckInputs
|
|
"test_is_copy_consistent"
|
|
]
|
|
++ lib.optionals (pythonAtLeast "3.12") [
|
|
|
|
# RuntimeError: Dynamo is not supported on Python 3.12+
|
|
"test_from_save_pretrained_dynamo"
|
|
];
|
|
|
|
meta = with lib; {
|
|
description = "State-of-the-art diffusion models for image and audio generation in PyTorch";
|
|
mainProgram = "diffusers-cli";
|
|
homepage = "https://github.com/huggingface/diffusers";
|
|
changelog = "https://github.com/huggingface/diffusers/releases/tag/${lib.removePrefix "refs/tags/" src.rev}";
|
|
license = licenses.asl20;
|
|
maintainers = with maintainers; [ natsukium ];
|
|
};
|
|
}
|