litgpt
litgpt copied to clipboard
Add chat mode for finetuned models using Adapter.
To chat with the finetuned model interactively using: python chat/adapter.py The default parameters: - model path: out/adapter/alpaca/lit_model_adapter_finetuned.pth - model type: stabilityai/stablelm-base-alpha-3b - finetune data: alpaca - finetune tech: Adapter Test success on NVIDIA DGX-A100.
This is a separate script added to the ./chat/ folder, won't affect the other script logic.