ipex-llm/python/llm/example/GPU/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 Update llm gpu xpu default related info to PyTorch 2.1 (#9866) 2024-01-09 15:38:47 +08:00
Save-Load Update llm gpu xpu default related info to PyTorch 2.1 (#9866) 2024-01-09 15:38:47 +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 GPU

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