-
Notifications
You must be signed in to change notification settings - Fork 1
/
models.rb
90 lines (80 loc) · 2.49 KB
/
models.rb
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
class Cloudflare::AI::Models
class << self
def all
{
automatic_speech_recognition: automatic_speech_recognition,
image_classification: image_classification,
image_to_text: image_to_text,
object_detection: object_detection,
summarization: summarization,
text_classification: text_classification,
text_embeddings: text_embedding,
text_generation: text_generation,
text_to_image: text_to_image,
translation: translation
}
end
def automatic_speech_recognition
%w[@cf/openai/whisper]
end
def image_classification
%w[@cf/microsoft/resnet-50]
end
def image_to_text
%w[@cf/unum/uform-gen2-qwen-500m]
end
def object_detection
%w[@cf/meta/detr-resnet-50]
end
def summarization
%w[@cf/facebook/bart-large-cnn]
end
def text_classification
%w[@cf/huggingface/distilbert-sst-2-int8]
end
def text_embedding
%w[@cf/baai/bge-base-en-v1.5 @cf/baai/bge-large-en-v1.5 @cf/baai/bge-small-en-v1.5]
end
def text_generation
%w[
@hf/thebloke/codellama-7b-instruct-awq
@hf/thebloke/deepseek-coder-6.7b-base-awq
@hf/thebloke/deepseek-coder-6.7b-instruct-awq
@cf/deepseek-ai/deepseek-math-7b-base
@cf/deepseek-ai/deepseek-math-7b-instruct
@cf/thebloke/discolm-german-7b-v1-awq
@cf/tiiuae/falcon-7b-instruct
@hf/thebloke/llama-2-13b-chat-awq
@cf/meta/llama-2-7b-chat-fp16
@cf/meta/llama-2-7b-chat-int8
@hf/thebloke/llamaguard-7b-awq
@cf/mistral/mistral-7b-instruct-v0.1
@hf/thebloke/mistral-7b-instruct-v0.1-awq
@hf/thebloke/neural-chat-7b-v3-1-awq
@hf/thebloke/openchat_3.5-awq
@cf/openchat/openchat-3.5-0106
@hf/thebloke/openhermes-2.5-mistral-7b-awq
@cf/microsoft/phi-2
@cf/qwen/qwen1.5-0.5b-chat
@cf/qwen/qwen1.5-1.8b-chat
@cf/qwen/qwen1.5-14b-chat-awq
@cf/qwen/qwen1.5-7b-chat-awq
@cf/defog/sqlcoder-7b-2
@cf/tinyllama/tinyllama-1.1b-chat-v1.0
@hf/thebloke/zephyr-7b-beta-awq
]
end
def text_to_image
%w[
@cf/lykon/dreamshaper-8-lcm
@cf/runwayml/stable-diffusion-v1-5-img2img
@cf/runwayml/stable-diffusion-v1-5-inpainting
@cf/stabilityai/stable-diffusion-xl-base-1.0
@cf/bytedance/stable-diffusion-xl-lightning
]
end
def translation
%w[@cf/meta/m2m100-1.2b]
end
end
end