AI开发平台MODELARTS-使用AI Gallery SDK构建自定义模型:后续操作

时间：2024-10-23 09:02:58

AI开发平台MODELARTS 构建模型

自定义模型文件构建完成后，可以参考托管模型到AI Gallery将模型文件托管至AI Gallery。建议托管的模型文件列表参见表2。

表2 模型实例包含的文件
文件名称	描述
config.json	模型配置文件。
model.safetensors或pytorch_model.bin	预训练模型的权重文件。
tokenizer.json	（可选）预处理器的词表文件，用于初始化Tokenizer。
tokenizer_config.json	（可选）预处理器的配置文件。
modeling_xxx.py	（可选）自定义模型的代码文件，继承自PretrainedModel，包含实现自定义推理逻辑的代码。
configuration_xxx.py	（可选）自定义配置的代码文件，继承自PretrainedConfig，包含实现自定义配置的逻辑代码。