Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
3.7 kB · 160 lines
Nix
at gcc-offload
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161{ 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 ]; };}