# environment.yml
name: allm # The name of the environment

channels: # The conda channels to search for packages
  # - pytorch
  - conda-forge
  # - dnachun
  # - anaconda
channel_priority: strict

dependencies:
  # Packages to install with conda
  - python=3.11.3
  #- pytorch-cuda=12.4
  #- pytorch
  # - numpy
  - transformers>=4.55
  - einops
  - jaxtyping

  - tensorboard
  - omegaconf
  - accelerate
  - peft

  - wandb

  - scipy
  - pandas
  - matplotlib
  - scikit-image
  - scikit-learn
  - joblib
  - pillow
  - datasets
  ## NO - huggingface_hub
  - tqdm
  - nltk
  - future
  

  - defusedxml
  - ipdb
  - torchinfo



  - timm
  - graphviz #anaconda::graphviz
  - dnachun::torchviz
  - pip:
  #  - draccus
     - fraction
     - vllm