NVIDIA / workbench-llamafactory

This is an NVIDIA AI Workbench example project that demonstrates an end-to-end model development workflow using Llamafactory.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

LLaMA-Factory - AI Workbench Project

This is an NVIDIA AI Workbench project to deploy LLaMA-Factory.

Please follow the LLM fine-tuning tutorial for RTX AI Toolkit here.

Minimum recommended Workbench ver: 0.50.16

Docker Desktop ver. 4.31+

Features

  • Fine-tune an LLM using Llama factory.
  • Fuse generated LoRA adapter with the base model.
  • Apply Post Training Quantization to your model.
  • Export model to TensorRT-LLM checkpoint.
  • Support for unsloth.

About

This is an NVIDIA AI Workbench example project that demonstrates an end-to-end model development workflow using Llamafactory.

License:Apache License 2.0


Languages

Language:Jupyter Notebook 73.1%Language:Shell 19.3%Language:Python 7.6%