File size: 542 Bytes
7feac49 |
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 |
# Copyright (c) Alibaba, Inc. and its affiliates.
BIN_EXTENSIONS = [
'.*.bin',
'.*.ts',
'.*.pt',
'.*.data-00000-of-00001',
'.*.onnx',
'.*.meta',
'.*.pb',
'.*.index',
]
PEFT_TYPE_KEY = 'peft_type'
SWIFT_TYPE_KEY = 'swift_type'
DEFAULT_ADAPTER = 'default'
class Invoke(object):
KEY = 'invoked_by'
THIRD_PARTY = 'third_party'
PRETRAINED = 'from_pretrained'
PIPELINE = 'pipeline'
TRAINER = 'trainer'
LOCAL_TRAINER = 'local_trainer'
PREPROCESSOR = 'preprocessor'
SWIFT = 'swift'
|