google / google/tunix

A unified API for model loading

Open
#612 1 comment 0 reactions 1 assignee Claimed by @tianshub View on GitHub
type:feature/enhancement
Dominant language
Python
Stars
2.5k
Forks
345
Avg merge
1d 7h
Merged PRs (30d)
240

Description

Currently, for loading different models in Tunix we need to load the model and its params_loader in a very hard python specific module loading (e.g. from tunix.models.qwen3.params import create_model_from_safe_tensors or from tunix.models.qwen3 import model as model_lib)

I think it is based on best practices that creating a higher level abstraction like AutoModel for model creation and loading it weights must be beneficial and unifying all the models under the same abstract AutoModel

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.