Blame view

environment.yml 894 Bytes
be52af70   tangwang   first commit
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
  name: searchengine
  channels:
    - pytorch
    - nvidia
    - conda-forge
    - defaults
  
  dependencies:
    - python=3.10
    - pip
    - pytorch>=2.0.0
    - torchvision
    - cudatoolkit=11.8
    - numpy>=1.24.0
    - pandas>=2.0.0
  
    # Install via pip
    - pip:
      # Core
      - pyyaml>=6.0
      - python-dotenv>=1.0.0
  
      # Database
      - pymysql>=1.1.0
      - sqlalchemy>=2.0.0
  
      # Elasticsearch
      - elasticsearch>=8.0.0,<9.0.0
  
      # Redis
      - redis>=5.0.0
  
      # ML/Embeddings
      - sentence-transformers>=2.2.0
      - transformers>=4.30.0
      - modelscope>=1.9.0
      - cn-clip>=1.5.0
      - pillow>=10.0.0
  
      # API
      - fastapi>=0.100.0
      - uvicorn[standard]>=0.23.0
      - pydantic>=2.0.0
      - python-multipart>=0.0.6
  
      # Translation
      - requests>=2.31.0
  
      # Utilities
      - tqdm>=4.65.0
      - click>=8.1.0
  
      # Testing
      - pytest>=7.4.0
      - pytest-asyncio>=0.21.0
      - httpx>=0.24.0