memory-framework/environment.yaml at main · naiimic/memory-framework · 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
name: gen
channels:
    - anaconda # IMPORTANT TO ADD THIS FOR sentence_transformers
    - conda-forge
    - defaults
dependencies:
    - python==3.9
    - pip
    - openai # OpenAI API
    - pytest
    - pytest-mock
    - pip:
          - hydra-core # Meta AI hydra job/configuration management
          - gymnasium # OpenAI/Farama Foundation Gym environment
          - protobuf<=3.20.x # For fixing conflict
          - mlagents-envs==0.30.0 # Unity ML-agents
          - seaborn # seaborn plotting
          - scikit-learn
          # For developers
          - pytest-mock
          - django-environ
          - shapely
          - geopandas
          - networkx
          - langchain # alternative LLM: LangChain (compatible with GPT and Vicuna)
          - faiss-cpu
          - google-generativeai # alternative LLM: Google PaLM
          - spacy
          - sentence_transformers
          - black
          - python-dotenv # For loading .env files
          - boto3 # For interaction with AWS
          - lightning  # PyTorch Lightning
          # Packages (maybe not the minimal set) to get sentence_transformer to work
          - huggingface_hub
          - numpy
          - pandas
          - pillow
          - psutil
          - scikit-learn
          - scipy
          - sentencepiece
          - tiktoken
          - tokenizers
          - torch
          - torchvision
          - tqdm
          - transformers