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

• Work in progress! (ALPHA)
• English | Русский | 漢語
• 

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:

StableDiffusion:

StableAudio:

Interface:

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

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

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

Linux

1. First install all RequiredDependencies
2. Git clone https://github.com/Dartvauder/NeuroTrainerWebUI.git to any location
3. In the terminal, run the ./Install.sh and wait for installation of all dependencies
4. After installation, run ./Start.sh
5. Wait for the application to launch
6. 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

Third Party Licenses:

Many models have their own license for use. Before using it, I advise you to familiarize yourself with them:

• Transformers
• Diffusers
• StableDiffusion1.5
• StableDiffusion2
• StableDiffusionXL
• CLIP
• BERT
• LLAMA.CPP
• StableAudioOpen
• BLIP

These third-party repository codes are also used in my project:

• Diffusers scripts for training SD
• StableAudioTools for training StableAudio

Donation

If you liked my project and want to donate, here is options to donate. Thank you very much in advance!

• 

Star History