cliobot/config.example.yml at main · herval/cliobot · 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
env: development
mode: command
locale: en

openai:
  models:
    - gpt-4
    - whisper-1
    - dall-e-3
    - gpt-4-vision

  endpoints:
    - api_key: $OPENAI_API_TOKEN
      api_type: open_ai
      base_url: https://api.openai.com/v1/

    - api_key: $OPENAI_AZURE_API_KEY
      api_type: azure
      api_version: 2023-10-01-preview
      base_url: $OPENAI_AZURE_BASE_URL
      model: gpt4
      kind: gpt-4

    - api_key: $OPENAI_AZURE_API_KEY
      api_type: azure
      api_version: 2
      base_url: $OPENAI_AZURE_BASE_URL
      model: embeddings
      kind: embeddings

    - api_key: $OPENAI_AZURE_API_KEY
      api_type: azure
      api_version: 2023-12-01-preview
      base_url: $OPENAI_AZURE_BASE_URL
      model: dalle3
      kind: dall-e-3

    - api_key: $OPENAI_AZURE_API_KEY
      api_type: azure
      api_version: 2023-12-01-preview
      base_url: $OPENAI_AZURE_BASE_URL
      model: whisper1
      kind: whisper-1


ollama:
  endpoint: http://localhost:11434
  models:
    - model: llama2
      kind: 'ask'
    - model: llava
      kind: 'describe'

replicate:
  api_token: $REPLICATE_API_TOKEN
  endpoints:
    - model: 'llava13'
      kind: 'describe'
      version: "yorickvp/llava-13b:6bc1c7bb0d2a34e413301fee8f7cc728d2d4e75bfab186aa995f63292bda92fc"
      params:
        prompt:
          kind: str
          required: true
        image:
          kind: image
          required: true
        max_tokens:
          kind: int
          default: 1024
    - model: 'sdxl'
      kind: 'image'
      version: 'stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b'
      params:
        prompt:
          kind: str
          required: true
        negative_prompt:
          alias: no
          kind: str
        width:
          kind: int
          default: 1024
        height:
          kind: int
          default: 1024
        num_outputs:
          alias: num
          kind: int
          default: 1
        num_inference_steps:
          alias: steps
          kind: int
          default: 25
        guidance_scale:
          alias: cfg
          kind: float
          default: 7.5
        prompt_strength:
          alias: ps
          kind: float
          default: 0.8
        seed:
          kind: int
        apply_watermark:
          alias: watermark
          kind: bool
          default: true
        scheduler:
          kind: str
          default: 'KarrasDPM'
        refine:
          kind: str
          alias: refiner
          default: 'no_refiner'
          value_map:
            no: no_refiner
            expert: expert_ensemble_refiner
            base: base_image_refiner
        refine_steps:
          kind: int
          alias: rs

bot:
  platform: telegram
  token: $TELEGRAM_TOKEN

db:
  #  driver: inmemory
  driver: sqlite3
  file: data/db.sqlite3

storage:
  driver: local
  folder: data/
#  driver: s3
#  bucket: my-bucket
#  zone: eu-west-1
#  access_key: my-key
# secret_key: my-secret


fallback_commands:
  audio: transcribe
  voice: transcribe
  text: ask


default_models:
  ask: llama2
  transcribe: whisper-1
  image: dall-e-3
  describe: gpt-4-vision-preview