tanmayshishodia
71d980267d
Why? this commit adds a generic finetuning pipeline with LoRA and DoRA support
21 lines
282 B
YAML
21 lines
282 B
YAML
name: tfm_env
|
|
|
|
channels:
|
|
- conda-forge
|
|
- defaults
|
|
- anaconda
|
|
dependencies:
|
|
- jupyterlab
|
|
- pip
|
|
- python=3.10
|
|
- pip:
|
|
- huggingface_hub[cli]
|
|
- utilsforecast
|
|
- praxis
|
|
- paxml
|
|
- jax[cpu]==0.4.26
|
|
- einshape
|
|
- scikit-learn
|
|
- typer
|
|
- wandb
|