python3Packages.accelerate: Bump version 1.0.0 -> 1.1.0 and fix few things (#353184)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user