自定义模型训练序列分类问题，problem type报错

[evantian110]

Describe the bug
当我自定义了一个模型（LLM+Linear）想训练seq_cls任务的时候，出现了报错。（由于是在内网部署的，不能截图只能尽可能的描述准确）
是template/base.py中 line 512
encoded = self._seq_cls_encode(chosen)
template/base.py中 line 465
problem_type = self.config.problem_type
NoneType object has no attribution problem_type

Your hardware and system info
用的pip install e . from source 安装

Additional context
在文件llm/model/register.py文件中 line 597 行 定义ModelInfo变量的时候没有传入config参数，导致后面的config为None值。
我自己的修改方式是ModelInfo新加了一个参数problem_type=None，
然后在把kwargs里面的problm_type参数设置为problem_type，
所以problem_type的值不在config里面取了，直接从modelinfo里面拿。

由于代码在内网中，无法描述的很清楚，希望可以看明白。

[Jintao-Huang]

请提供一下训练脚本

[evantian110]

请提供一下训练脚本

CUDA_VISIVLE_DEVICES=0
swift set
--model /path/to/Qwen3-0.6B
--train_full full
--dataset /path/to/train.jsonl
--torch_dtype bfloat16
--num_train_epochs 1
--per_device_train_batch_size 1
--learning_rate 1e-4
--output_dir /path/to/output \
--task_type seq_cls \
--num_labels 2
--problem_type single_label_classification \
--template template_example \
--custom_register_path register_path_example.py \
--model_type model_type_example

*_example我自己随便自定义的名字

[Jintao-Huang]

ms-swift/swift/llm/model/register.py

Line 299 in f278882

model_config.problem_type = problem_type

[evantian110]

ms-swift/swift/llm/model/register.py

Line 299 in f278882

model_config.problem_type = problem_type

函数get_model_tokenizer_from_local()在register.py文件中只出现了两个地方，一个是get_model_tokenizer_with_flash_attn，由于我自定义了get_model_tokenizer_func，所以我进不去这个。还有一个就是line 646处。
if model_meta is None:
model_meta = ModelMeta(None, [], 'dummy', get_model_tokenizer_from_local, model_arch=None)
logger.info(f'Temporarily create model_meta: {model_meta}')
由于命令行中定义了model_type，以及在注册文件中运行了
register_model(
ModelMeta(
model_type='model_type_example', # 和命令行中的一致
model_groups=[],
template='template_exampl',
get_function=get_function,# 自定义的
is_multimodal=False,
)
)
所以modelmeta不会是None，所以没办法执行get_model_tokenizer_from_local。是因为我注册文件里面写错了吗？

[Jin-zd]

我也在自定义模型进行序列分类任务时遇到过同样的 “‘NoneType’ object has no attribute ‘problem_type’” 报错，原因应该是用自定义 register 的 get_function 时，Template 读的是 model_info.config.problem_type，但 model_info.config 根本没被填，等于 None，就报错了。
我在自定义的 get_function 中加载好模型的 config 后添加下面的代码：

# 将问题类型写入 config，避免模板在 seq_cls encode 时读取到 None
setattr(config, "problem_type", problem_type)
# 将 config 回写到 model_info，供 Template.init_processor 使用
if hasattr(model_info, 'config'):
    model_info.config = config

Template._seq_cls_encode 就不会再从 None 里读 problem_type，训练能正常启动了。