cbb3a09dd8
- 在 tools 目录下新增 model.py 文件 - 实现 get_model_name 函数，用于获取模型的名称 - 更新 tools/__init__.py，导入新的 get_model_name 函数
4 lines
85 B
Python
4 lines
85 B
Python
import os
|
|
|
|
def get_model_name(model):
|
|
return os.path.basename(model.name_or_path) |