Welcome to AI Forge! This repository offers a simple yet powerful Python system designed for training and conversing with custom-trained models using CSV files. Whether you're a beginner eager to explore AI or an expert looking for a flexible tool to craft bespoke conversational models, AI Forge provides the framework to build and interact with your unique AI systems effortlessly.
- Train custom models on .CSV based corpora
- Converse with your custom models
- VRAM Usage Estimation Algorithm
- Interactive and intuitive user interface
- Supports multiple languages and dialects
- Customizable training parameters for fine-tuning models
- Real-time conversation capabilities
- Detailed documentation and examples
- Export training models for use in other applications
- Lightweight and efficient, suitable for deployment on low-resource devices
- Community-driven support and contributions
Python 3.10 or newer
- Clone the repository:
git clone https://github.com/ConnerAdamsMaine/AIForge
cd AIForge
- Create and activate a virtual environment (optional but recommended):
python -m venv venv
source venv/bin/activate # On windows use 'venv\Scripts\activate'
- Install the required libraries:
pip install -r requirements.txt
- Pandas for handling the CSV files
- Numpy
- Skikit-Learn
- Tensorflow
- Keras
- Prepare your training data in a CSV file.
- Use the provided scripts to train your model.
- Start a conversation with your trained model.
from AIForge import Trainer
trainer = Trainer('path_to_your_csv_file.csv')
model = await trainer.Train()from AIForge import Conversationalist
conversationalist = Conversationalist(model)
response = await conversationalist.converse('Your input here')
print(response)