diff --git a/recipe/meta.yaml b/recipe/meta.yaml index fac1da6..621184f 100644 --- a/recipe/meta.yaml +++ b/recipe/meta.yaml @@ -1,5 +1,5 @@ {% set name = "deepspeed" %} -{% set version = "0.11.1" %} +{% set version = "0.11.2" %} {% 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: fa342f11f971e1bb002a537adfea6ac4c089b7a4da5aef5f2216a472cb35be7f + sha256: 6d6f716882a3a92ebea37aaeec05a063dc0fe1445d3ab2fc2b603ba87a8a2300 build: number: {{ number }} @@ -66,7 +66,8 @@ requirements: - psutil - py-cpuinfo # PyDantic <2.0.0 is required until DeepSpeed updates to be compatible with newer PyDantic versions. - - pydantic <2.0.0 + - pydantic + - pynvml - python - pytorch # [cuda_compiler_version in (undefined, 'None')] - tqdm