Skip to content

Commit

Permalink
Update supported_models.yaml
Browse files Browse the repository at this point in the history
Signed-off-by: Ishaan Sehgal <ishaanforthewin@gmail.com>
  • Loading branch information
ishaansehgal99 authored Jan 23, 2024
1 parent c9a63bc commit f43334c
Showing 1 changed file with 59 additions and 10 deletions.
69 changes: 59 additions & 10 deletions presets/models/supported_models.yaml
Original file line number Diff line number Diff line change
@@ -1,11 +1,60 @@
models:
- falcon-7b
- falcon-7b-instruct
- falcon-40b
- falcon-40b-instruct
- llama-2-7b
- llama-2-7b-chat
- llama-2-13b
- llama-2-13b-chat
- llama-2-70b
- llama-2-70b-chat
# Llama
- name: llama-2-7b
type: text-generation
runtime: llama
tag: 0.0.1
- name: llama-2-7b-chat
type: text-generation
runtime: llama
tag: 0.0.1
- name: llama-2-13b
type: text-generation
runtime: llama
tag: 0.0.1
- name: llama-2-13b-chat
type: text-generation
runtime: llama
tag: 0.0.1
- name: llama-2-70b
type: text-generation
runtime: llama
tag: 0.0.1
- name: llama-2-70b-chat
type: text-generation
runtime: llama
tag: 0.0.1

# Falcon
- name: falcon-7b
type: text-generation
version: 898df1396f35e447d5fe44e0a3ccaaaa69f30d36
runtime: tfs
tag: 0.0.1
- name: falcon-7b-instruct
type: text-generation
version: cf4b3c42ce2fdfe24f753f0f0d179202fea59c99
runtime: tfs
tag: 0.0.1
- name: falcon-40b
type: text-generation
version: 4a70170c215b36a3cce4b4253f6d0612bb7d4146
runtime: tfs
tag: 0.0.1
- name: falcon-40b-instruct
type: text-generation
version: ecb78d97ac356d098e79f0db222c9ce7c5d9ee5f
runtime: tfs
tag: 0.0.1

# Mistral
- name: mistral-7b-v0.1
type: text-generation
version: 26bca36bde8333b5d7f72e9ed20ccda6a618af24
runtime: tfs
tag: 0.0.1
- name: mistral-7b-instruct-v0.1
type: text-generation
version: 9ab9e76e2b09f9f29ea2d56aa5bd139e4445c59e
runtime: tfs
tag: 0.0.1

0 comments on commit f43334c

Please sign in to comment.