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 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