Open-Ended-Goal-Decoding/environment_for_DalEyeLlama.yml at main · lacclab/Open-Ended-Goal-Decoding · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
name: DalEyeLlama
channels:
  - pytorch
  - huggingface
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - python=3.12.8
  - pandas=2.2.3
  - seaborn=0.13
  - matplotlib=3.10.0
  - numpy=2.2.2
  # - transformers=4.47.1 # Careful as models assumes this version (could work with others but not tested)
  - transformers=4.53.2 # Changed from original version for unsloth compatibility
  - pytorch=2.5.1
  - pytorch-cuda=11.8
  - pyarrow
  - pyyaml
  - pynvml # for checking gpu utilization
  - datasets=3.2.0
  - lightning=2.5.0
  - scikit-learn=1.6.1
  - pip
  - wordfreq=3.1
  - wandb=0.19
  - pre-commit
  - ruff
  - hydra-core=1.3
  - ipykernel
  - beartype>=0.17
  - mypy
  - pytorch-metric-learning=2.8.1 #=2.4
  - torchmetrics=1.6.1 #.3.1 # 1.3.2 falls on 'ClasswiseWrapper' object has no attribute 'preds'
  - pandas-stubs # for type checking
  - types-tqdm # for type checking
  - py-xgboost
  - openai
  - gpustat
  - ipywidgets
  - accelerate
  - typed-argument-parser
  - peft=0.15
  # added for running notebooks/question-decoding/LLM_on_scanpath_syntax.py
  - loguru=0.7.2
  - trl=0.12.1

  - pip:
      - -e . # For development purposes
      - protobuf==3.20.3  # Pin protobuf to compatible version for ML libraries. Not the same as Cfir did
      - tap
      - rich
      - juliacall
      - git+https://github.com/getzze/statannotations@compat-seaborn-13
      - bert-score
      # added for running notebooks/question-decoding/LLM_on_scanpath_syntax.py
      - unsloth
      - sentence-transformers