Features | Dependencies | SystemRequirements | Install | Models | Wiki | Acknowledgment | Licenses

project-image

Work in progress! (ALPHA)
English | Русский | 漢語
<a href='https://github.com/MShawon/github-clone-count-badge'><img alt='GitHub Clones' src='https://img.shields.io/badge/dynamic/json?color=success&label=Clone&query=count&url=https://gist.githubusercontent.com/Dartvauder/aa704ab4803f8694f2436cbf874a34fd/raw/clone.json&logo=github'></a>

Description:

A simple and convenient interface for using of various neural network models. You can create datasets, finetune, evaluate and generate with LLM, StableDiffusion and StableAudio using various hyperparameters. You can also check the wiki, download the LLM, StableDiffusion and StableAudio models, change the application settings inside the interface and check system sensors

The goal of the project - to create the easiest possible application to finetune, evaluate and generate of neural network models

LLM: <img width="1114" alt="1" src="https://github.com/user-attachments/assets/ed89c506-8b1d-49bb-8579-d1fdcfb94d9f">

StableDiffusion: <img width="1111" alt="2" src="https://github.com/user-attachments/assets/19e21251-8fd1-4007-8ff6-7cfe1fff8f68">

StableAudio: <img width="1116" alt="3" src="https://github.com/user-attachments/assets/d549715b-fb57-42c8-82c3-d34ad0977ec0">

Interface: <img width="1111" alt="4" src="https://github.com/user-attachments/assets/5ac059af-5951-44b4-9fa4-85d1d8d83e42">

Features:

Easy installation via install.bat(Windows) or install.sh(Linux)
You can use the application via your mobile device in localhost (Via IPv4) or anywhere online (Via Share)
Flexible and optimized interface (By Gradio)
Debug logging to logs from Install and Update files
Available in three languages
Support for Transformers: finetune, evaluate, quantize and generate (LLM)
Support for Diffusers and Safetensors: finetune, evaluate, conversion, quantize and generate (StableDiffusion)
Support for StableAudio: finetune and generate
Full and LORA types of finetune, evaluate and generate (For LLM and StableDiffusion)
Ability to create a dataset (For LLM, StableDiffusion and StableAudio models)
Wiki
ModelDownloader
Application settings
Ability to see system sensors

Required Dependencies:

Python (3.10.11)
Git
CUDA (12.4) and cuDNN (9.1)

C+ compiler
- Windows: VisualStudio, VisualStudioCode and Cmake
- Linux: GCC, VisualStudioCode and Cmake

Minimum System Requirements:

System: Windows or Linux
GPU: 8GB+ or CPU: 16 core 3.6Ghz
RAM: 24GB+
Disk space: 10GB+
Internet for installing

How to install:

Windows

First install all RequiredDependencies
Git clone https://github.com/Dartvauder/NeuroTrainerWebUI.git to any location
Run the Install.bat and wait for installation
After installation, run Start.bat
Wait for the application to launch
Now you can start generating!

To get update, run Update.bat To work with the virtual environment through the terminal, run Venv.bat

Linux

First install all RequiredDependencies
Git clone https://github.com/Dartvauder/NeuroTrainerWebUI.git to any location
In the terminal, run the ./Install.sh and wait for installation of all dependencies
After installation, run ./Start.sh
Wait for the application to launch
Now you can start generating!

To get update, run ./Update.sh To work with the virtual environment through the terminal, run ./Venv.sh

Wiki

https://github.com/Dartvauder/NeuroTrainerWebUI/wiki

Acknowledgment to developers

Many thanks to these projects because thanks to their applications/libraries, i was able to create my application:

First of all, I want to thank the developers of PyCharm and GitHub. With the help of their applications, i was able to create and share my code

gradio - https://github.com/gradio-app/gradio
transformers - https://github.com/huggingface/transformers
diffusers - https://github.com/huggingface/diffusers
llama.cpp-python - https://github.com/abetlen/llama-cpp-python
stable-diffusion-cpp-python - https://github.com/william-murray1204/stable-diffusion-cpp-python
stable-audio-tools - https://github.com/Stability-AI/stable-audio-tools

NeuroTrainerWebUI

Install / Use

README