This repository has been archived by the owner on Jun 26, 2024. It is now read-only.
Chatbot support & bug fixes
What's Changed
- Chatbot by @aniketmaurya in #4
- Refactor bot by @aniketmaurya in #5
How to use Chatbot
from chatbot import LLaMAChatBot
checkpoint_path = f"state_dict.pth"
tokenizer_path = f"tokenizer.model"
bot = LLaMAChatBot(
checkpoint_path=checkpoint_path, tokenizer_path=tokenizer_path
)
print(bot.send("hi, what is the capital of France?"))
Full Changelog: v0.0.2...v0.0.3