environment.yml 894 Bytes
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