diff --git a/recipe/meta.yaml b/recipe/meta.yaml index 621184f..81a8322 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -1,5 +1,5 @@ {% set name = "deepspeed" %} -{% set version = "0.11.2" %} +{% set version = "0.12.1" %} {% set number = 0 %} {% set torch_proc_type = "cuda" if cuda_compiler_version != "None" else "cpu" %} @@ -15,7 +15,7 @@ package: source: url: https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/{{ name }}-{{ version }}.tar.gz - sha256: 6d6f716882a3a92ebea37aaeec05a063dc0fe1445d3ab2fc2b603ba87a8a2300 + sha256: 4b15201a004aed2e32d37d251ce02f74c09980ddc020a2a84ea28adac076cb68 build: number: {{ number }} @@ -58,6 +58,7 @@ requirements: - pytorch # [cuda_compiler_version not in (undefined, 'None')] - pytorch =*={{ torch_proc_type }}* # [cuda_compiler_version not in (undefined, 'None')] run: + - ninja - hjson-py # NOTE(hadim): coming at https://github.com/conda-forge/staged-recipes/pull/19098 # - ninja-python