fasr-lid-firered 0.5.0

FireRedLID language identification model for fasr

fasr-lid-firered

基于 FireRedLID 的语种识别模型插件，为 fasr 提供音频语种自动识别能力。

安装

pip install fasr-lid-firered

注册模型

注册名	类	默认 checkpoint	说明
firered	FireRedForLID	xukaituo/FireRedLID	语种识别，返回语言标识（如 "zh"、"en"）

模型权重默认从 ModelScope 自动下载。

使用方式

单独使用模型

模型实例化时会自动执行 download_checkpoint() + load_checkpoint()：

from fasr.config import registry
from fasr.data import Waveform

model = registry.lid_models.get("firered")(use_gpu=True, use_half=True)

waveform = Waveform.from_file("example.wav")
lang = model.identify(waveform)
print(lang)  # e.g. "zh"

# 使用自定义权重目录
model.load_checkpoint("/path/to/custom/firered-lid")

运行期 / 会话参数

参数	类型	默认值	说明
checkpoint	str | None	"FireRedTeam/FireRedLID"	远程 repo_id；非空时实例化会自动下载
cache_dir	str | Path | None	None	缓存目录，None 使用 fasr.utils.get_cache_dir()
endpoint	Literal["modelscope", "huggingface", "hf-mirror"]	"modelscope"	下载端点
use_gpu	bool	True	是否使用 GPU 推理
use_half	bool	False	是否使用 FP16 推理加速
max_chunk_seconds	float	60.0	单次推理最大音频时长；更长音频会被自动切块

依赖

• fasr
• torch >= 2.0.0
• kaldiio >= 2.18.0、kaldi-native-fbank >= 1.19.0
• Python 3.10–3.12