| # 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' | |