Python pretrained-models

Open-source Python projects categorized as pretrained-models

Top 23 Python pretrained-model Projects

  • transformers

    🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

    Project mention: AI enthusiasm #6 - Finetune any LLM you want💡 | | 2024-04-16

    Most of this tutorial is based on Hugging Face course about Transformers and on Niels Rogge's Transformers tutorials: make sure to check their work and give them a star on GitHub, if you please ❤️

  • pytorch-image-models

    PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

    Project mention: FLaNK AI Weekly 18 March 2024 | | 2024-03-18
  • InfluxDB

    Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  • spleeter

    Deezer source separation library including pretrained models.

    Project mention: Are stems a good way of making mashups | /r/Beatmatch | 2023-12-10

    virtual dj and others stem separator is shrinked model of this you will get better results downloading original + their large model.

  • PaddleNLP

    👑 Easy-to-use and powerful NLP and LLM library with 🤗 Awesome model zoo, supporting wide-range of NLP tasks from research to industrial applications, including 🗂Text Classification, 🔍 Neural Search, ❓ Question Answering, ℹ️ Information Extraction, 📄 Document Intelligence, 💌 Sentiment Analysis etc.

  • Qwen

    The official repo of Qwen (通义千问) chat & pretrained large language model proposed by Alibaba Cloud.

    Project mention: What the heck is so great about this model? | /r/SillyTavernAI | 2023-12-07


  • segmentation_models.pytorch

    Segmentation models with pretrained backbones. PyTorch.

    Project mention: Instance segmentation of small objects in grainy drone imagery | /r/computervision | 2023-12-09

    Also, I’d suggest considering switching to the segmentation-models library - it provides U-Net models with a variety of pretrained backbones of as encoders. The author also put out a PyTorch version.

  • petals

    🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading

    Project mention: Mistral Large | | 2024-02-26

    So how long until we can do an open source Mistral Large?

    We could make a start on Petals or some other open source distributed training network cluster possibly?


  • WorkOS

    The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  • LMFlow

    An Extensible Toolkit for Finetuning and Inference of Large Foundation Models. Large Models for All.

    Project mention: Your weekly machine learning digest | /r/learnmachinelearning | 2023-07-03
  • CodeGeeX

    CodeGeeX: An Open Multilingual Code Generation Model (KDD 2023)

    Project mention: For Developers - THUDM/CodeGeeX: CodeGeeX: An Open Multilingual Code Generation Model | /r/OfflineAI | 2023-05-20
  • EfficientNet-PyTorch

    A PyTorch implementation of EfficientNet and EfficientNetV2 (coming soon!)

  • mmf

    A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

  • PaddleClas

    A treasure chest for visual classification and recognition powered by PaddlePaddle

  • CogVLM

    a state-of-the-art-level open visual language model | 多模态预训练模型

    Project mention: Mixtral: Mixture of Experts | | 2024-01-08

    CogVLM is very good in my (brief) testing:

    The model weights seem to be under a non-commercial license, not true open source, but it is "open access" as you requested.

  • awesome-pretrained-chinese-nlp-models

    Awesome Pretrained Chinese NLP Models,高质量中文预训练模型&大模型&多模态模型&大语言模型集合

  • facenet-pytorch

    Pretrained Pytorch face detection (MTCNN) and facial recognition (InceptionResnet) models

    Project mention: [D] Fast face recognition over video | /r/MachineLearning | 2023-04-22

    Hijacking this comment because i've been working nonstop on my project thanks to your suggestion. I'm now using this for face detection and still the old face_recognition for encodings. I'm clustering with dbscan and extracting frames with ffmpeg with -hwaccel on. I'm planning to try this: as it looks like it would be the fastest thing avaiable to process videos? Keep in mind i need to perform encoding other than just detection because i want to use DBscan (and later also facial recognition, but this might be done separately just by saving the encodings). let me know if you have any other suggestions, and thanks again for your help

  • Efficient-AI-Backbones

    Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

  • Chinese-CLIP

    Chinese version of CLIP which achieves Chinese cross-modal retrieval and representation generation.

  • PINTO_model_zoo

    A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

  • mmpretrain

    OpenMMLab Pre-training Toolbox and Benchmark

  • Pretrained-Language-Model

    Pretrained language model and its related optimization techniques developed by Huawei Noah's Ark Lab.

    Project mention: Does anyone know a downloadable chatgpt model that supports conversation in Albanian? | /r/Programimi | 2023-05-16
  • deepsparse

    Sparsity-aware deep learning inference runtime for CPUs

    Project mention: Fast Llama 2 on CPUs with Sparse Fine-Tuning and DeepSparse | | 2023-11-23

    Interesting company. Yannic Kilcher interviewed Nir Shavit last year and they went into some depth: DeepSparse is on GitHub:

  • OFA

    Official repository of OFA (ICML 2022). Paper: OFA: Unifying Architectures, Tasks, and Modalities Through a Simple Sequence-to-Sequence Learning Framework

  • asteroid

    The PyTorch-based audio source separation toolkit for researchers

  • SaaSHub

    SaaSHub - Software Alternatives and Reviews. SaaSHub helps you find the best software and product alternatives

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020). The latest post mention was on 2024-04-16.

Python pretrained-models related posts


What are some of the best open-source pretrained-model projects in Python? This list will help you:

Project Stars
1 transformers 124,115
2 pytorch-image-models 29,659
3 spleeter 24,839
4 PaddleNLP 11,335
5 Qwen 10,691
6 segmentation_models.pytorch 8,773
7 petals 8,631
8 LMFlow 7,975
9 CodeGeeX 7,732
10 EfficientNet-PyTorch 7,715
11 mmf 5,414
12 PaddleClas 5,244
13 CogVLM 4,896
14 awesome-pretrained-chinese-nlp-models 4,172
15 facenet-pytorch 4,129
16 Efficient-AI-Backbones 3,774
17 Chinese-CLIP 3,529
18 PINTO_model_zoo 3,262
19 mmpretrain 3,137
20 Pretrained-Language-Model 2,953
21 deepsparse 2,866
22 OFA 2,318
23 asteroid 2,089
SaaSHub - Software Alternatives and Reviews
SaaSHub helps you find the best software and product alternatives