Class: Cloudflare::AI::Models

Inherits:
Object
  • Object
show all
Defined in:
lib/cloudflare/ai/models.rb

Class Method Summary collapse

Class Method Details

.all ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
14
15
16
 
# File 'lib/cloudflare/ai/models.rb', line 3

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

.automatic_speech_recognition ⇒ Object 



18
19
20
 
# File 'lib/cloudflare/ai/models.rb', line 18

def automatic_speech_recognition
  %w[@cf/openai/whisper]
end

.image_classification ⇒ Object 



22
23
24
 
# File 'lib/cloudflare/ai/models.rb', line 22

def image_classification
  %w[@cf/microsoft/resnet-50]
end

.image_to_text ⇒ Object 



26
27
28
 
# File 'lib/cloudflare/ai/models.rb', line 26

def image_to_text
  %w[@cf/unum/uform-gen2-qwen-500m]
end

.object_detection ⇒ Object 



30
31
32
 
# File 'lib/cloudflare/ai/models.rb', line 30

def object_detection
  %w[@cf/meta/detr-resnet-50]
end

.summarization ⇒ Object 



34
35
36
 
# File 'lib/cloudflare/ai/models.rb', line 34

def summarization
  %w[@cf/facebook/bart-large-cnn]
end

.text_classification ⇒ Object 



38
39
40
 
# File 'lib/cloudflare/ai/models.rb', line 38

def text_classification
  %w[@cf/huggingface/distilbert-sst-2-int8]
end

.text_embedding ⇒ Object 



42
43
44
 
# File 'lib/cloudflare/ai/models.rb', line 42

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

.text_generation ⇒ Object 



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
 
# File 'lib/cloudflare/ai/models.rb', line 46

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

.text_to_image ⇒ Object 



76
77
78
79
80
81
82
83
84
 
# File 'lib/cloudflare/ai/models.rb', line 76

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

.translation ⇒ Object 



86
87
88
 
# File 'lib/cloudflare/ai/models.rb', line 86

def translation
  %w[@cf/meta/m2m100-1.2b]
end