From 5d24f14c83387ab82e55cea4df72c694c642a9d1 Mon Sep 17 00:00:00 2001 From: Karl Date: Thu, 30 Jan 2025 13:20:01 -0600 Subject: [PATCH 1/2] ADD mps env file --- mps_environment.yml | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 mps_environment.yml diff --git a/mps_environment.yml b/mps_environment.yml new file mode 100644 index 0000000..6566496 --- /dev/null +++ b/mps_environment.yml @@ -0,0 +1,18 @@ +name: mps-lm-training +channels: + - pytorch + - defaults +dependencies: + - python=3.10.* + - pip + - pip: + - black + - mypy + - torch==2.* + - transformers==4.* + - accelerate==1.0.* + - evaluate==0.4.* + - jiwer==3.* + - sentencepiece==0.* # Omit if not working with models requiring SentencePiece + - hydra-core==1.3.* + - strenum From 842a314d0f9d47f5efe776f82e3ff540a8bd4b70 Mon Sep 17 00:00:00 2001 From: Karl Date: Fri, 14 Feb 2025 10:53:28 -0600 Subject: [PATCH 2/2] CHG existing env names to distinguish CUDA from MPS --- environment.yml => cpu_environment.yml | 0 gpu_environment.yml => cuda_environment.yml | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) rename environment.yml => cpu_environment.yml (100%) rename gpu_environment.yml => cuda_environment.yml (96%) diff --git a/environment.yml b/cpu_environment.yml similarity index 100% rename from environment.yml rename to cpu_environment.yml diff --git a/gpu_environment.yml b/cuda_environment.yml similarity index 96% rename from gpu_environment.yml rename to cuda_environment.yml index c72bf26..9df8f4a 100644 --- a/gpu_environment.yml +++ b/cuda_environment.yml @@ -1,4 +1,4 @@ -name: gpu-lm-training +name: cuda-lm-training channels: - pytorch - nvidia