forked from predictionguard/mpt-7b-truss
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathconfig.yaml
32 lines (32 loc) · 881 Bytes
/
config.yaml
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
bundled_packages_dir: packages
data_dir: data
description: MPT-7B is an open-source AI language model developed by MosaicML. MPT-7B Instruct is a model for short-form instruction following.
environment_variables: {}
examples_filename: examples.yaml
input_type: Any
live_reload: false
model_class_filename: model.py
model_class_name: Model
model_framework: custom
model_metadata:
avatar_url: https://avatars.githubusercontent.com/u/75143706
cover_image_url: https://assets-global.website-files.com/61e6fd1bd5174813bfe40bba/64499b1820d2b969b273c45c_butterfly_c.png
tags:
- text-generation
model_module_dir: model
model_name: MPT-7B Instruct
model_type: custom
python_version: py39
requirements:
- transformers
- torch
- einops
- sentencepiece
resources:
cpu: "3"
memory: 50Gi
use_gpu: true
accelerator: T4
secrets: {}
spec_version: '2.0'
system_packages: []