CodonTranslator / environment.yml
alegendaryfish's picture
Align public training codebase with local training setup
0f643b6 verified
name: codontranslator
channels:
- conda-forge
- pytorch
- nvidia
dependencies:
- python=3.12
- pip
- pytorch>=2.4
- pandas>=2.3
- pyarrow>=21.0
- duckdb>=1.5
- biopython>=1.85
- requests>=2.32
- tqdm>=4.67
- pip:
- transformers>=4.57.0
- esm>=3.2.3
- safetensors>=0.7.0
- huggingface-hub>=0.36.0
- accelerate>=1.9.0
- wandb>=0.21.0