flow.yml

jtype: Flow
with:
  protocol: grpc
  port: 51005
  env:
    JINA_LOG_LEVEL: debug
executors:
  - name: dalle
    uses: executors/dalle/config.yml
    env:
      CUDA_VISIBLE_DEVICES: 2
  - name: diffusion
    uses: GLID3Diffusion
    uses_with:
      glid3_path: ../glid-3-xl
      steps: 100
    py_modules:
      - executors/glid3/executor.py
    env:
      CUDA_VISIBLE_DEVICES: 1
    replicas: 1
    needs: [gateway]
  - name: rerank
    uses: ReRank
    uses_with:
      clip_server: grpc://0.0.0.0:51000
    py_modules:
      - executors/rerank/executor.py
    needs: [dalle, diffusion]
  - name: upscaler
    uses: SwinIRUpscaler
    py_modules:
      - executors/swinir/executor.py
    uses_with:
      swinir_path: ../SwinIR
      store_path: dalle.db
    env:
      CUDA_VISIBLE_DEVICES: 1
    replicas: 2
#  - name: store
#    uses: MyStore
#    py_modules:
#      - executors/store/executor.py
#    uses_with:
#      store_path: dalle-flow.db