python3Packages.accelerate: Bump version 1.0.0 -> 1.1.0 and fix few things (#353184) (6ee487a5) · Commits · nix / nixpkgs

pkgs/development/python-modules/accelerate/default.nix

+29 −2

Original line number	Diff line number	Diff line
		@@ -12,6 +12,7 @@
		setuptools,

		# dependencies
		huggingface-hub,
		numpy,
		packaging,
		psutil,
		@@ -30,14 +31,14 @@

		buildPythonPackage rec {
		pname = "accelerate";
		version = "1.0.0";
		version = "1.1.0";
		pyproject = true;

		src = fetchFromGitHub {
		owner = "huggingface";
		repo = "accelerate";
		rev = "refs/tags/v${version}";
		hash = "sha256-XVJqyhDSUPQDHdaB6GDxHhuC6EWCSZNArjzyLpvhQHI=";
		hash = "sha256-GBNe4zomy8dmfvYrk/9Q77Z6r+JJA+2dgAhJx2opqAc=";
		};

		buildInputs = [ llvmPackages.openmp ];
		@@ -45,6 +46,7 @@ buildPythonPackage rec {
		build-system = [ setuptools ];

		dependencies = [
		huggingface-hub
		numpy
		packaging
		psutil
		@@ -113,6 +115,31 @@ buildPythonPackage rec {
		"test_init_trackers"
		"test_log"
		"test_log_with_tensor"

		# After enabling MPS in pytorch, these tests started failing
		"test_accelerated_optimizer_step_was_skipped"
		"test_auto_wrap_policy"
		"test_autocast_kwargs"
		"test_automatic_loading"
		"test_backward_prefetch"
		"test_can_resume_training"
		"test_can_resume_training_checkpoints_relative_path"
		"test_can_resume_training_with_folder"
		"test_can_unwrap_model_fp16"
		"test_checkpoint_deletion"
		"test_cpu_offload"
		"test_cpu_ram_efficient_loading"
		"test_grad_scaler_kwargs"
		"test_invalid_registration"
		"test_map_location"
		"test_mixed_precision"
		"test_mixed_precision_buffer_autocast_override"
		"test_project_dir"
		"test_project_dir_with_config"
		"test_sharding_strategy"
		"test_state_dict_type"
		"test_with_save_limit"
		"test_with_scheduler"
		]
		++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [
		# RuntimeError: torch_shm_manager: execl failed: Permission denied