ipex-llm/python/llm/example/CPU/PyTorch-Models
Zheng, Yi 2347f611cf Add cpu and gpu examples of Mamba (#9797)
* Add mamba cpu example

* Add mamba gpu example

* Use a smaller model as the example

* minor fixes

---------

Co-authored-by: Shengsheng Huang <shengsheng.huang@intel.com>
2024-02-28 11:33:29 +08:00
..
Model Add cpu and gpu examples of Mamba (#9797) 2024-02-28 11:33:29 +08:00
More-Data-Types LLM: add CPU More-Data-Types and Save-Load examples (#9179) 2023-10-17 14:38:52 +08:00
Save-Load LLM: add CPU More-Data-Types and Save-Load examples (#9179) 2023-10-17 14:38:52 +08:00
README.md LLM: update example layout (#9046) 2023-10-09 15:36:39 +08:00

Running PyTorch model using BigDL-LLM on Intel CPU

This folder contains examples of running any PyTorch model on BigDL-LLM (with "one-line code change"):

  • Model: examples of running PyTorch models (e.g., Openai Whisper, LLaMA2, ChatGLM2, Falcon, MPT, Baichuan2, etc.) using INT4 optimizations
  • More-Data-Types: examples of applying other low bit optimizations (NF4/INT5/INT8, etc.)
  • Save-Load: examples of saving and loading low-bit models